yolov5 / utils /torch_utils.py

Commit History

Update torch_utils.py (#1895)
ffef771
unverified

glenn-jocher commited on

Add colorstr() (#1887)
6ab5895
unverified

glenn-jocher commited on

YOLOv5 v4.0 Release (#1837)
69be8e7
unverified

glenn-jocher commited on

Torch CUDA synchronize update (#1826)
9f5a18b
unverified

glenn-jocher commited on

Display correct CUDA devices (#1776)
c0ffcdf
unverified

glenn-jocher commited on

Automatic m.half() profile on x.half()
3004fb5
unverified

glenn-jocher commited on

Update torch_utils.py
0bd9c48
unverified

glenn-jocher commited on

Input channel yaml['ch'] addition (#1741)
394d1c8
unverified

glenn-jocher commited on

Update loss criteria constructor (#1711)
8bc0027
unverified

glenn-jocher commited on

Fix torch multi-GPU --device error (#1701)
035ac82
unverified

Nanobit glenn-jocher commited on

Profile() feature addition (#1673)
ada90e3
unverified

glenn-jocher commited on

Increase FLOPS robustness (#1608)
8918e63
unverified

glenn-jocher commited on

FLOPS computation device bug fix (#1447)
05a955a
unverified

yujun glenn-jocher commited on

Improved FLOPS computation (#1398)
9f27902

glenn-jocher commited on

Improved FLOPS computation (#1398)
d8f5fcf
unverified

glenn-jocher commited on

Utils reorganization (#1392)
fe341fa
unverified

glenn-jocher commited on

PyTorch version to screen and cleanup (#1325)
19e2482
unverified

glenn-jocher commited on

PyTorch 1.7.0 Compatibility Updates (#1233)
c8c5ef3
unverified

glenn-jocher commited on

Simplified inference (#1153)
3b57cb5
unverified

glenn-jocher commited on

torchvision nms bug fix
124f0e8

glenn-jocher commited on

classifier, export, torch seed updates
883924d

glenn-jocher commited on

.fuse() gradient introduction bug fix
89655a8

glenn-jocher commited on

add support for DWConv fuse operation (#898)
c5969f7
unverified

tkianai commited on

Update torch_utils.py (#751)
26c3b11
unverified

lesscomfortable commited on

Fix redundant outputs via Logging in DDP training (#500)
4949401
unverified

Nanobit glenn-jocher commited on

Single-source training (#680)
3c6e2f7
unverified

glenn-jocher commited on

PyTorch 1.6.0 update with native AMP (#573)
c020875
unverified

goodfellowliu commited on

update yolo.py TTA flexibility and extensibility (#506)
1d17b9a
unverified

glenn-jocher commited on

v2.0 Release (#491)
9da56b6
unverified

glenn-jocher commited on

update torch_utils.py to FP16 EMA
41ff83a

glenn-jocher commited on

multi-gpu ckpt filesize bug fix #253
a586751

glenn-jocher commited on

--resume EMA fix #292
24c5a94

glenn-jocher commited on

update fuse_conv_and_bn()
121d90b

glenn-jocher commited on

pruning and sparsity initial commit
38f5c1a

glenn-jocher commited on

Update to remove "module" attribute check
ccfa937
unverified

Nanobit commited on

Fix save error for multi-gpu
f83521a
unverified

Nanobit commited on

Update torch_utils.py
f02481c
unverified

glenn-jocher commited on

There is no need to download extra packages, official bring it with you
8b8b792

goodfellowliu commited on

augmented inference
b810b21

glenn-jocher commited on

speed-reproducibility fix #17
22d6088

glenn-jocher commited on