10 Commits (c5f5b80c04585a9df6dcde88c072dbff5ed1cf90)

Author SHA1 Message Date
Ayush Chaurasia c5f5b80c04
Deterministic training (#53) 2 years ago
ayush chaurasia 793dde365d Revert "update" 2 years ago
ayush chaurasia 486bc49414 update 2 years ago
Ayush Chaurasia d7df1770fa
Add warmup and accumulation (#52) 2 years ago
Ayush Chaurasia 4291b9c31c
Add EMA and model checkpointing (#49) 2 years ago
Glenn Jocher 27d6545117
General console printout updates (#48) 2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40) 3 years ago
Ayush Chaurasia 1054819a59
Add initial model interface (#30) 3 years ago
Ayush Chaurasia 7b560f7861
Model builder (#29) 3 years ago
Ayush Chaurasia d0b3c9812b
Trainer + Dataloaders (#27) 3 years ago