Glenn Jocher
|
5e2429e618
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
7bd7b2c090
|
update export.py
|
4 years ago |
Glenn Jocher
|
ce0c58f678
|
update compute_loss()
|
4 years ago |
Glenn Jocher
|
af41083d52
|
EMA FP16 fix #279
|
4 years ago |
Glenn Jocher
|
f767023c56
|
offset and balance update
|
4 years ago |
Glenn Jocher
|
655895a838
|
module updates
|
4 years ago |
Glenn Jocher
|
df224a0d8f
|
EMA bug fix #279
|
4 years ago |
Glenn Jocher
|
a9d20eba3e
|
export.py update
|
4 years ago |
Glenn Jocher
|
569fbc63c8
|
export.py update
|
4 years ago |
Glenn Jocher
|
3bdea3f697
|
strip_optimizer() bug fix #253
|
4 years ago |
Glenn Jocher
|
6ca3f35cd4
|
update .dockerignore
|
4 years ago |
Glenn Jocher
|
e02a189a3a
|
Merge pull request #245 from yxNONG/patch-2
Unify the check point of single and multi GPU
|
4 years ago |
Glenn Jocher
|
f02481c73a
|
Update torch_utils.py
|
4 years ago |
Glenn Jocher
|
fc7c42723d
|
Update utils.py
|
4 years ago |
Glenn Jocher
|
597ed4ce63
|
Update train.py
|
4 years ago |
Glenn Jocher
|
13f69777a6
|
typo fix
|
4 years ago |
yxNONG
|
1aa2b67933
|
Update train.py
|
4 years ago |
yxNONG
|
53cdaf6bf5
|
Update utils.py
|
4 years ago |
Glenn Jocher
|
a03adb5ab8
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
1fca7a7f24
|
autopad() update in common.py
|
4 years ago |
Glenn Jocher
|
5323ad224d
|
export.py update
|
4 years ago |
Glenn Jocher
|
a62a1c2c67
|
export.py update
|
4 years ago |
Glenn Jocher
|
f1d67f4110
|
update export.py
|
4 years ago |
Glenn Jocher
|
e74ccb2985
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
b5659d1195
|
module updates
|
4 years ago |
Glenn Jocher
|
dfd63de20a
|
Merge pull request #255 from edurenye/ignore_torchscript
Add torchscript files to gitignore
|
4 years ago |
edurenye
|
1c86c2cb66
|
Add torchscript files to gitignore
|
4 years ago |
Glenn Jocher
|
86784cfdbf
|
--resume bug fix #252
|
4 years ago |
Glenn Jocher
|
0a08375a8a
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
4e62eb90be
|
check_img_size() int() update
|
4 years ago |
Glenn Jocher
|
5c1c73ecda
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
ad4c22cbfe
|
--resume bug fix #187
|
4 years ago |
Glenn Jocher
|
2b7a11300e
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
a9dc0c2c29
|
update check_img_size() for model strides
|
4 years ago |
Glenn Jocher
|
b9da3ea46a
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
3b16c865f0
|
assert --epochs
|
4 years ago |
yxNONG
|
cdb9bde181
|
Unify the check point of single and multi GPU
save the model.hyp etc to checkpoint when use multi GPU training
|
4 years ago |
Glenn Jocher
|
bfd51f62f8
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
9a9333d245
|
add replicate() to datasets.py
|
4 years ago |
Glenn Jocher
|
615d6d0cfa
|
FROM nvcr.io/nvidia/pytorch:20.06-py3
|
4 years ago |
Glenn Jocher
|
e8bf385795
|
Merge pull request #223 from ArmyOfRobots/master
Adding torchscript export
|
4 years ago |
Glenn Jocher
|
b203c9b7ff
|
update train.py incompatible model message fix #222
|
4 years ago |
Glenn Jocher
|
01895f3924
|
Merge pull request #228 from topiaruss/patch-1
Remove non-ascii that breaks pdb single-stepping
|
4 years ago |
Russ Ferriday
|
0d5065d3b9
|
Remove non-ascii that breaks pdb single-stepping
|
4 years ago |
Glenn Jocher
|
0bc80e1e57
|
onnx_export.py bug fix #224
|
4 years ago |
Glenn Jocher
|
ec8554a39b
|
FROM nvcr.io/nvidia/pytorch:20.06-py3
|
4 years ago |
Jake Poznanski
|
685091a3fc
|
Adding torchscript export
|
4 years ago |
Glenn Jocher
|
37e13f8846
|
update mosaic border
|
4 years ago |
Glenn Jocher
|
93a6765806
|
update mosaic border
|
4 years ago |
Glenn Jocher
|
77fb8ee082
|
refactor dataloader
|
4 years ago |