6 Commits (298287298d330482ce1a7bfae7fdf18565f2971c)

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