22 Commits

Author SHA1 Message Date
Laughing
3a241e4cea
update segment training (#57)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: ayush chaurasia <ayush.chaurarsia@gmail.com>
2022-11-29 17:00:08 +05:30
Ayush Chaurasia
c5f5b80c04
Deterministic training (#53)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-25 02:06:14 +05:30
ayush chaurasia
793dde365d Revert "update"
This reverts commit 486bc494142744e9d90f0483b611a785381401a7.
2022-11-25 01:54:50 +05:30
ayush chaurasia
486bc49414 update 2022-11-25 01:11:55 +05:30
Ayush Chaurasia
d7df1770fa
Add warmup and accumulation (#52)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-25 00:42:16 +05:30
Ayush Chaurasia
298287298d
Add clearml logging (#51)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-24 15:25:03 +01:00
Ayush Chaurasia
4291b9c31c
Add EMA and model checkpointing (#49)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-19 19:07:26 +01:00
Glenn Jocher
27d6545117
General console printout updates (#48)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 16:08:16 +01:00
Laughing
47f1cb3ef4
Fix some cuda training issues of segmentation (#46)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-17 18:14:02 +05:30
Ayush Chaurasia
db1031a1a9
Allow setting model attributes before training (#45) 2022-11-17 10:35:53 +05:30
Ayush Chaurasia
832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-15 15:36:29 +01:00
Ayush Chaurasia
1f3aad86c1
new check_dataset functions (#43)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-12 15:02:41 +01:00
Glenn Jocher
4c68b9dcf6
Add is_colab() and is_kaggle() (#41)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-08 17:14:49 +01:00
Ayush Chaurasia
f56c9bcc26
Segmentation support & other enchancements (#40)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-08 16:27:57 +01:00
Glenn Jocher
e4f7458d90
Hydra *.yml extension deprecated fix (#34) 2022-11-07 00:52:08 +01:00
Glenn Jocher
6fe8bead35
Classify training cleanup (#33)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-07 00:15:57 +01:00
Ayush Chaurasia
92c60758dd
Smart Model loading (#31)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-31 23:52:12 +01:00
Ayush Chaurasia
1054819a59
Add initial model interface (#30)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-25 21:51:15 +02:00
Ayush Chaurasia
7b560f7861
Model builder (#29)
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-10-18 23:14:23 +02:00
Ayush Chaurasia
c5cb76b356
Metrics and loss structure (#28)
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-10-15 19:39:05 +02:00
Ayush Chaurasia
d0b3c9812b
Trainer + Dataloaders (#27)
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ayush Chaurasia <ayushchaurasia@Ayushs-MacBook-Pro.local>
Co-authored-by: Ayush Chaurasia <ayush.chuararsia@gmail.com>
2022-10-10 14:01:07 +02:00
Ayush Chaurasia
419f6d7167
Add yolo package structure (#15) 2022-09-19 15:47:23 +05:30