Glenn Jocher
41523e2c91
Dataset autodownload feature addition ( #685 )
...
* initial commit
* move download scripts into data/scripts
* new check_dataset() function in general.py
* move check_dataset() out of with context
* Update general.py
* DDP update
* Update general.py
5 years ago
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
5 years ago
Glenn Jocher
43a616a955
PyTorch 1.6.0 compatability updates
5 years ago
Glenn Jocher
07a82f4d44
Update ci-testing.yml ( #445 )
...
* Update ci-testing.yml
* Update ci-testing.yml
* Update requirements.txt
* Update requirements.txt
* Update google_utils.py
* Update test.py
* Update ci-testing.yml
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