58 Commits (efc941aa8131eac680872edf202b8c23be176abd)

Author SHA1 Message Date
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