YOLO / train.py

Commit History

✨ [Init] Trainer for training whole model!
649c592

henry000 commited on

πŸ”¨ [Add] Dataset to train.py, todo: Dataloader
95ca62f

henry000 commited on

🎨 [Update] pre-commit module, add isort
31cab2b

henry000 commited on

🚸 [Add] auto download option for download dataset
1ff26a6

henry000 commited on

🚚 [Rename] tools and utils, move the function!
d5ba31a

henry000 commited on

πŸ”¨ [Add] Auto download dataset before training
461c87f

henry000 commited on

πŸ”€ [Merge] branch 'MODEL' into SETUP
787b81d

henry000 commited on

πŸ”¨ [Change] argparse to hydra, faster-easier!
5fed6e3

henry000 commited on

πŸ”‡ [Remove] the log for whole model in train.py
bb5c520

henry000 commited on

πŸ§‘β€πŸ’» [add] custom logger, new format of logging
642e32f

henry000 commited on

πŸ”¨ [Init] the base train code, and load yaml cfg
b80dc1e

henry000 commited on