11 Commits (d0b0fe2592225ccc71132c47d61d16435105d644)

Author SHA1 Message Date
Ayush Chaurasia d0b0fe2592
CLI updates (#58)
2 years ago
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
Glenn Jocher 6fe8bead35
Classify training cleanup (#33)
2 years ago
Ayush Chaurasia 1054819a59
Add initial model interface (#30)
2 years ago
Ayush Chaurasia c5cb76b356
Metrics and loss structure (#28)
2 years ago
Ayush Chaurasia d0b3c9812b
Trainer + Dataloaders (#27)
2 years ago