6 Commits (c5f5b80c04585a9df6dcde88c072dbff5ed1cf90)

Author SHA1 Message Date
Ayush Chaurasia c5f5b80c04
Deterministic training (#53)
2 years ago
Ayush Chaurasia d7df1770fa
Add warmup and accumulation (#52)
2 years ago
Ayush Chaurasia 512a225ce8
Cli support (#50)
2 years ago
Ayush Chaurasia 832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40)
2 years ago
Glenn Jocher e4f7458d90
Hydra *.yml extension deprecated fix (#34)
2 years ago