173 Commits (8f5eeb09f39d9f18c02c7fc2deded812d9a1f1b0)

Author SHA1 Message Date
Ayush Chaurasia c5f5b80c04
Deterministic training (#53)
2 years ago
ayush chaurasia 793dde365d Revert "update"
2 years ago
ayush chaurasia 486bc49414 update
2 years ago
Ayush Chaurasia d7df1770fa
Add warmup and accumulation (#52)
2 years ago
Ayush Chaurasia 298287298d
Add clearml logging (#51)
2 years ago
Ayush Chaurasia 512a225ce8
Cli support (#50)
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
Glenn Jocher 8530e3fae0
Logging fix from YOLOv5 (#47)
2 years ago
Ayush Chaurasia 832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
2 years ago
Ayush Chaurasia 1f3aad86c1
new check_dataset functions (#43)
2 years ago
Glenn Jocher 4c68b9dcf6
Add is_colab() and is_kaggle() (#41)
2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40)
2 years ago
Laughing c617ee1c79
Fix dataloader2 (#35)
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
Laughing 2e9b18ce4e
Fix dataloader (#32)
2 years ago
Ayush Chaurasia 92c60758dd
Smart Model loading (#31)
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 c5cb76b356
Metrics and loss structure (#28)
2 years ago
Ayush Chaurasia d0b3c9812b
Trainer + Dataloaders (#27)
2 years ago
Ayush Chaurasia 419f6d7167
Add yolo package structure (#15)
2 years ago