A basic and simple training framework for pytorch. Easy to expand for multi network interaction
[TOC] Introduction A basic and simple training framework for pytorch, easy for extension. Dependence torch==1.7.0+cu110 torchvision==0.8.0 easydict==1.9 tensorboard==2.7 tensorboardX==2.4 PyYAML==5.3.1 Features Any module can be easily customized Not abstract, easy to learn, develop and debug With a lot of repetitive work reduction, it can be more easily control the training process in development and research Friendly to multi-network interactive training, such as GAN, transfer learning, knowledge distillation, etc. Train python main.py –model resnet18 –save_dir cifar100_resnet18 –config_path ./configs/cifar100.yml Results CIFAR100 All […]
Read more