summaryrefslogtreecommitdiff
path: root/config.py
AgeCommit message (Collapse)Author
2021-01-14Enable optimizer fine tuningJordan Gong
2021-01-13Update config file and convert int to str when joiningJordan Gong
2021-01-13Add multiple checkpoints for different model and set default config valueJordan Gong
2021-01-12Some changes in hyperparameter configJordan Gong
1. Separate hyperparameter configs in model, optimizer and scheduler 2. Add more tunable hyperparameters in optimizer and scheduler
2021-01-07Change device config and add enable multi-GPU computingJordan Gong
1. Add `disable_acc` switch for disabling accelerator. When it is off, system will automatically choosing accelerator. 2. Enable multi-GPU training using torch.nn.DataParallel
2021-01-03Update hyperparameter configuration, implement prototype fit functionJordan Gong
2020-12-27Change default dataset directoryJordan Gong
2020-12-27Make naming scheme consistentJordan Gong
Use `dir` instead of `path`
2020-12-26Add config file and corresponding type hintJordan Gong