58 Commits (9f482cbcb885c7b37df47285afd3e3bbaaaa780d)

Author SHA1 Message Date
Glenn Jocher 41523e2c91
Dataset autodownload feature addition (#685)
5 years ago
Jirka Borovec d5b6416c87
Explicit Imports (#498)
5 years ago
Glenn Jocher 43a616a955 PyTorch 1.6.0 compatability updates
5 years ago
Glenn Jocher 07a82f4d44
Update ci-testing.yml (#445)
5 years ago
Glenn Jocher b40852d5a5 update test.py
5 years ago
Glenn Jocher 37acbdc0b6 update test.py --save-txt
6 years ago
Glenn Jocher b8c2da4a0d update test.py --save-txt
6 years ago
Glenn Jocher 1ce686edfc update test.py
6 years ago
Glenn Jocher 6bd9218922 update test.py
6 years ago
Glenn Jocher aa08b2bd6f test.py --task test updates
6 years ago
Glenn Jocher a1c8406af3 EMA and non_blocking=True
6 years ago
Glenn Jocher 0afbb8d498 update test.py model.half()
6 years ago
Glenn Jocher bf6f41567a hyperparameter printout update
6 years ago
Glenn Jocher dc5e18390a
Merge branch 'master' into advanced_logging
6 years ago
Glenn Jocher 22ab1c295f
Update test.py
6 years ago
Glenn Jocher c3d3e6b776
Update test.py
6 years ago
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318
6 years ago
Alex Stoken 5ac517b22a Remove duplicate `verbose` arg in test.py
6 years ago
Alex Stoken f517ba81c7
Merge branch 'master' into advanced_logging
6 years ago
Glenn Jocher 6b95d6d2c0 .to(device) bug fix
6 years ago
Glenn Jocher a62333e808 test.py .fuse() update
6 years ago
Glenn Jocher 04bdbe4104 fuse update
6 years ago
Glenn Jocher a9dc0c2c29 update check_img_size() for model strides
6 years ago
Alex Stoken e18e6811dc
Merge branch 'master' into advanced_logging
6 years ago
Glenn Jocher 22fb2b0c25 refactor dataloader
6 years ago
Glenn Jocher e670a3353b update test.py for FP16 testing during training
6 years ago
Glenn Jocher 227aa735dc updates
6 years ago
Glenn Jocher b8557f87e3 add stride to datasets.py
6 years ago
Glenn Jocher a63e1c9b3c study update
6 years ago
Glenn Jocher db898131f9 study update
6 years ago
Glenn Jocher 6c1b87a42e update google_utils import
6 years ago
Glenn Jocher 1f1917ef56 remove fast, add merge
6 years ago
Alex Stoken 9b7386f603 Add save_dir arg to test.test, use arg as location for saving batch jpgs
6 years ago
Glenn Jocher 2daa412575 train with multi-gpu half test bug fix #99
6 years ago
Glenn Jocher d1e5716610 train with multi-gpu half test bug fix #99
6 years ago
Glenn Jocher 72d06147b2 Multi-GPU disabled in test.py
6 years ago
Glenn Jocher 7c6fd4d670 --verbose bug fix #84
6 years ago
Glenn Jocher db2c3acd3a updated testing settings, rebalanced towards FP16 latency
6 years ago
Glenn Jocher 0b514daced FP16 test loss bug fix
6 years ago
Glenn Jocher a1748a8d6e test during training default to FP16
6 years ago
Glenn Jocher 76ca367a01 FP16 to FP32 ckpt load
6 years ago
Glenn Jocher c5966abba8 glob search bug fix #77
6 years ago
Glenn Jocher e7c24e04ac half() fix #66
6 years ago
Glenn Jocher 111994956e switch default inference to FP16 on GPU
6 years ago
Glenn Jocher 157aff2854 FP16 inference fix
6 years ago
Glenn Jocher 260b1729f0 FP16 inference update
6 years ago
Glenn Jocher 099e6f5ebd --img-size stride-multiple verification
6 years ago
Glenn Jocher d187459dd4 study changes
6 years ago
Glenn Jocher 546f973562 model fuse
6 years ago
Glenn Jocher 208493d959 updates
6 years ago