-
- Downloads
adding autocast
Showing
- data/config/loader/basic.yaml 1 addition, 1 deletiondata/config/loader/basic.yaml
- data/config/training/basic.yaml 6 additions, 10 deletionsdata/config/training/basic.yaml
- modules/training/training.py 31 additions, 20 deletionsmodules/training/training.py
- modules/utils/metric_aggregator.py 2 additions, 2 deletionsmodules/utils/metric_aggregator.py
- runs/01_train_model.py 4 additions, 21 deletionsruns/01_train_model.py
Loading
Please register or sign in to comment