6 Commits (e629335f6d2c9b2c43dbd718101a87239c367ef4)

Author SHA1 Message Date
Ayush Chaurasia 5a52e7663a
standalone val (#56)
2 years ago
Ayush Chaurasia 4291b9c31c
Add EMA and model checkpointing (#49)
2 years ago
Ayush Chaurasia 832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
2 years ago
Glenn Jocher 4c68b9dcf6
Add is_colab() and is_kaggle() (#41)
2 years ago
Ayush Chaurasia 1054819a59
Add initial model interface (#30)
2 years ago
Ayush Chaurasia 7b560f7861
Model builder (#29)
2 years ago