123456789101112131415161718192021 |
- '''Copyright The Microsoft DeepSpeed Team'''
- def check_tb_availability():
- try:
- # torch.utils.tensorboard will fail if `tensorboard` is not available,
- # see their docs for more details: https://pytorch.org/docs/1.8.0/tensorboard.html
- import tensorboard # noqa: F401
- except ImportError:
- print('If you want to use tensorboard logging, please `pip install tensorboard`')
- raise
- def check_wandb_availability():
- try:
- import wandb # noqa: F401
- except ImportError:
- print(
- 'If you want to use wandb logging, please `pip install wandb` and follow the instructions at https://docs.wandb.ai/quickstart'
- )
- raise
|