You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

__init__.py 346 B

5 years ago
12345678910111213
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. from tensorlayer.backend import BACKEND
  4. if BACKEND == 'tensorflow':
  5. from .tensorflow_metric import *
  6. elif BACKEND == 'mindspore':
  7. from .mindspore_metric import *
  8. elif BACKEND == 'paddle':
  9. from .paddle_metric import *
  10. else:
  11. raise NotImplementedError("This backend is not supported")

TensorLayer3.0 是一款兼容多种深度学习框架为计算后端的深度学习库。计划兼容TensorFlow, Pytorch, MindSpore, Paddle.