60 Commits (78840988573cce2af6505b026c40cee38749680a)

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)
2 years ago
Ayush Chaurasia 1054819a59
Add initial model interface (#30)
2 years ago
Ayush Chaurasia 7b560f7861
Model builder (#29)
2 years ago
Ayush Chaurasia d0b3c9812b
Trainer + Dataloaders (#27)
2 years ago