Spaces:

xiang-wuu
/

yolov5

Runtime error

App Files Files Community

yolov5 / utils /torch_utils.py

Commit History

Torch CUDA synchronize update (#1826)

9f5a18b
unverified

glenn-jocher commited on Jan 3, 2021

Display correct CUDA devices (#1776)

c0ffcdf
unverified

glenn-jocher commited on Dec 24, 2020

Automatic m.half() profile on x.half()

3004fb5
unverified

glenn-jocher commited on Dec 21, 2020

Update torch_utils.py

0bd9c48
unverified

glenn-jocher commited on Dec 21, 2020

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

394d1c8
unverified

glenn-jocher commited on Dec 19, 2020

Update loss criteria constructor (#1711)

8bc0027
unverified

glenn-jocher commited on Dec 16, 2020

Fix torch multi-GPU --device error (#1701)

035ac82
unverified

glenn-jocher commited on Dec 16, 2020

Profile() feature addition (#1673)

ada90e3
unverified

glenn-jocher commited on Dec 11, 2020

Increase FLOPS robustness (#1608)

8918e63
unverified

glenn-jocher commited on Dec 5, 2020

FLOPS computation device bug fix (#1447)

05a955a
unverified

yujun

glenn-jocher commited on Nov 19, 2020

Improved FLOPS computation (#1398)

9f27902

glenn-jocher commited on Nov 14, 2020

Improved FLOPS computation (#1398)

d8f5fcf
unverified

glenn-jocher commited on Nov 14, 2020

Utils reorganization (#1392)

fe341fa
unverified

glenn-jocher commited on Nov 14, 2020

PyTorch version to screen and cleanup (#1325)

19e2482
unverified

glenn-jocher commited on Nov 9, 2020

PyTorch 1.7.0 Compatibility Updates (#1233)

c8c5ef3
unverified

glenn-jocher commited on Oct 28, 2020

Simplified inference (#1153)

3b57cb5
unverified

glenn-jocher commited on Oct 15, 2020

torchvision nms bug fix

124f0e8

glenn-jocher commited on Oct 6, 2020

classifier, export, torch seed updates

883924d

glenn-jocher commited on Oct 6, 2020

.fuse() gradient introduction bug fix

89655a8

glenn-jocher commited on Sep 20, 2020

add support for DWConv fuse operation (#898)

c5969f7
unverified

tkianai commited on Sep 2, 2020

optimize imports

b42e8a5

glenn-jocher commited on Aug 21, 2020

Update torch_utils.py (#751)

26c3b11
unverified

lesscomfortable commited on Aug 16, 2020

reformat code

f5da528

glenn-jocher commited on Aug 14, 2020

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

4949401
unverified

glenn-jocher commited on Aug 11, 2020

Single-source training (#680)

3c6e2f7
unverified

glenn-jocher commited on Aug 9, 2020

PyTorch 1.6.0 update with native AMP (#573)

c020875
unverified

goodfellowliu commited on Jul 31, 2020

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

1d17b9a
unverified

glenn-jocher commited on Jul 24, 2020

v2.0 Release (#491)

9da56b6
unverified

glenn-jocher commited on Jul 23, 2020

FP16 EMA bug fix

2377e5f

glenn-jocher commited on Jul 13, 2020

update torch_utils.py to FP16 EMA

41ff83a

glenn-jocher commited on Jul 12, 2020

multi-gpu ckpt filesize bug fix #253

a586751

glenn-jocher commited on Jul 11, 2020

--resume EMA fix #292

24c5a94

glenn-jocher commited on Jul 9, 2020

update fuse_conv_and_bn()

121d90b

glenn-jocher commited on Jul 6, 2020

pruning and sparsity initial commit

38f5c1a

glenn-jocher commited on Jul 5, 2020

Update to remove "module" attribute check

ccfa937
unverified

Nanobit commited on Jul 4, 2020

Fix save error for multi-gpu

f83521a
unverified

Nanobit commited on Jul 4, 2020

EMA FP16 fix #279

af41083

glenn-jocher commited on Jul 3, 2020

EMA bug fix #279

df224a0

glenn-jocher commited on Jul 3, 2020

Update torch_utils.py

f02481c
unverified

glenn-jocher commited on Jul 2, 2020

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

8b8b792

goodfellowliu commited on Jun 16, 2020

model fuse

8fe299f

glenn-jocher commited on Jun 7, 2020

augmented inference

b810b21

glenn-jocher commited on Jun 6, 2020

updates

c12023f

glenn-jocher commited on Jun 6, 2020

speed-reproducibility fix #17

22d6088

glenn-jocher commited on Jun 5, 2020

initial commit

1e84a23

glenn-jocher commited on May 30, 2020