Spaces:

xiang-wuu
/

yolov5

Runtime error

App Files Files Community

yolov5 / utils /torch_utils.py

Commit History

Update DDP for `torch.distributed.run` with `gloo` backend (#3680)

fad27c0
unverified

glenn-jocher commited on Jun 19, 2021

Multi-GPU default to single device 0 (#3554)

8b5086c
unverified

glenn-jocher commited on Jun 9, 2021

Merge `develop` branch into `master` (#3518)

f3c3d2c
unverified

glenn-jocher Piotr Skalski

SkalskiP Peretz Cohen tudoulei chocosaj BuildTools

developer0hye Sam_S Samridha Shrestha edificewang commited on Jun 8, 2021

TensorBoard DP/DDP graph fix (#3325)

aad99b6
unverified

glenn-jocher commited on May 25, 2021

check `batch_size % utilized_device_count` (#3276)

f340235
unverified

glenn-jocher commited on May 21, 2021

Speed profiling improvements (#2648)

866bc7d
unverified

glenn-jocher commited on Mar 29, 2021

Improve git_describe() fix 1 (#2635)

2e95cf3
unverified

glenn-jocher commited on Mar 28, 2021

Improve git_describe() (#2633)

6e8c5b7
unverified

glenn-jocher commited on Mar 28, 2021

Create date_modified() (#2616)

0ff5aec
unverified

glenn-jocher commited on Mar 26, 2021

Update git_describe() for remote dir usage (#2606)

77415a4
unverified

glenn-jocher commited on Mar 26, 2021

PyTorch Hub models default to CUDA:0 if available (#2472)

9b11f0c
unverified

glenn-jocher commited on Mar 15, 2021

Integer printout (#2450)

f01f322
unverified

glenn-jocher commited on Mar 12, 2021

FLOPS min stride 32 (#2276)

83dc1b4
unverified

xiaowo1996 commited on Feb 23, 2021

Update torch_utils.py (#1895)

ffef771
unverified

glenn-jocher commited on Jan 10, 2021

Add colorstr() (#1887)

6ab5895
unverified

glenn-jocher commited on Jan 9, 2021

YOLOv5 v4.0 Release (#1837)

69be8e7
unverified

glenn-jocher commited on Jan 5, 2021

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