58 Commits (e21428ca4e0ed7f1d8fd8abad728989644b6a300)

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