Pytorch-Base-Trainer

Pytorch-Base-Trainer copied to clipboard

最新版0.8.1的问题

1

1.不断打印torch版本号 INFO: {'lr': 0.01, 'train': {'Accuracy': 98.828125, 'loss': 0.034419890493154526}} Train Epoch #4: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:42

GraceKafuu

混合精度训练

1

老哥，框架支持混合精度+DDP训练吗？

zephyrize