97 Commits (66497700010d0d759007f7c81f5948f53747996b)

Author SHA1 Message Date
Glenn Jocher 6649770001 --rect test bug fix
5 years ago
Glenn Jocher 520f5de6f0 Label caching foundational re-write #306
5 years ago
Glenn Jocher dd8e742ece
Merge pull request #334 from lorenzomammana/feature-multiple-datasets-training
5 years ago
Glenn Jocher dd33d2ab77
Update datasets.py
5 years ago
Glenn Jocher 2110f5ee89 update log_dir to runs/exp #107
5 years ago
Glenn Jocher 603ea0bfdc update log_dir to runs/exp #107
5 years ago
Glenn Jocher 24c5a941f0 --resume EMA fix #292
5 years ago
Glenn Jocher 13a09fa68a
Update datasets.py
5 years ago
Glenn Jocher 47bf1730fa
Update datasets.py
5 years ago
Glenn Jocher f9bab6b127
Update datasets.py
5 years ago
lorenzomammana 54a9e4f876 Refactor code to reduce duplication
5 years ago
lorenzomammana 94342acbe1 Handle multiple datasets
5 years ago
Glenn Jocher bf6f41567a hyperparameter printout update
5 years ago
Glenn Jocher dc5e18390a
Merge branch 'master' into advanced_logging
5 years ago
Glenn Jocher cbe39a1dbb
Update utils.py
5 years ago
Glenn Jocher c654d18f00
Update utils.py
5 years ago
Glenn Jocher 31b07b3550
Merge pull request #275 from wanghaoyang0106/master
5 years ago
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318
5 years ago
Alex Stoken 95f0a56df7 Bug fix to get_latest_run when recent run is named
5 years ago
Alex Stoken f517ba81c7
Merge branch 'master' into advanced_logging
5 years ago
Glenn Jocher 121d90b3f2 update fuse_conv_and_bn()
5 years ago
Glenn Jocher 38f5c1ad1d pruning and sparsity initial commit
5 years ago
NanoCode012 ccfa9373ae
Update to remove "module" attribute check
5 years ago
NanoCode012 f83521ab9a
Fix save error for multi-gpu
5 years ago
Glenn Jocher ce0c58f678 update compute_loss()
5 years ago
Glenn Jocher af41083d52 EMA FP16 fix #279
5 years ago
Glenn Jocher f767023c56 offset and balance update
5 years ago
Glenn Jocher df224a0d8f EMA bug fix #279
5 years ago
wanghaoyang0106 b57f83d005 [bug fix] potential problem if img fed to model is in rectangular shape
5 years ago
Glenn Jocher f02481c73a
Update torch_utils.py
5 years ago
Glenn Jocher fc7c42723d
Update utils.py
5 years ago
yxNONG 53cdaf6bf5
Update utils.py
5 years ago
Glenn Jocher 9a9333d245 add replicate() to datasets.py
5 years ago
Glenn Jocher 37e13f8846 update mosaic border
5 years ago
Glenn Jocher 93a6765806 update mosaic border
5 years ago
Alex Stoken e18e6811dc
Merge branch 'master' into advanced_logging
5 years ago
Glenn Jocher 77fb8ee082 refactor dataloader
5 years ago
Glenn Jocher 22fb2b0c25 refactor dataloader
5 years ago
Glenn Jocher b8557f87e3 add stride to datasets.py
5 years ago
Glenn Jocher 3b062254a6 update utils.create_pretrained()
5 years ago
Glenn Jocher 6c1b87a42e update google_utils import
5 years ago
Glenn Jocher 364fcfd7db PANet update
5 years ago
Glenn Jocher fc171e2678 check_anchor_order() update
5 years ago
Glenn Jocher 1f1917ef56 remove fast, add merge
5 years ago
Alex Stoken e572bb0803 Add plot_results save location to log_dir
5 years ago
Glenn Jocher 024a42f4fd datasets.py vid_formats updated
5 years ago
Glenn Jocher 5c2cd711df vid_formats bug fix #146
5 years ago
Glenn Jocher 10145833c3 strip_optimizer() default to save FP16 model
5 years ago
Glenn Jocher cce95e744d backbone as FP16, save default to FP32
5 years ago
Glenn Jocher 9fdb0fbacf AutoAnchor bug fix # 117
5 years ago