53 Commits (2dfe32030ad5f73d08275b93c0baa089bd513cf3)

Author SHA1 Message Date
  Glenn Jocher 0ff5aeca61
Create date_modified() (#2616) 3 years ago
  Glenn Jocher 77415a42e5
Update git_describe() for remote dir usage (#2606) 3 years ago
  Glenn Jocher 9b11f0c58b
PyTorch Hub models default to CUDA:0 if available (#2472) 3 years ago
  Glenn Jocher f01f3223d5
Integer printout (#2450) 3 years ago
  xiaowo1996 83dc1b4484
FLOPS min stride 32 (#2276) 3 years ago
  Glenn Jocher ffef77124e
Update torch_utils.py (#1895) 3 years ago
  Glenn Jocher 6ab589583c
Add colorstr() (#1887) 3 years ago
  Glenn Jocher 69be8e738f
YOLOv5 v4.0 Release (#1837) 3 years ago
  Glenn Jocher 9f5a18bb80
Torch CUDA synchronize update (#1826) 3 years ago
  Glenn Jocher c0ffcdf998
Display correct CUDA devices (#1776) 3 years ago
  Glenn Jocher 3004fb5bc1
Automatic m.half() profile on x.half() 3 years ago
  Glenn Jocher 0bd9c48609
Update torch_utils.py 3 years ago
  Glenn Jocher 394d1c89f3
Input channel yaml['ch'] addition (#1741) 3 years ago
  Glenn Jocher 8bc0027afc
Update loss criteria constructor (#1711) 3 years ago
  NanoCode012 035ac82ed0
Fix torch multi-GPU --device error (#1701) 3 years ago
  Glenn Jocher ada90e3901
Profile() feature addition (#1673) 3 years ago
  Glenn Jocher 8918e63476
Increase FLOPS robustness (#1608) 3 years ago
  yujun 05a955a3f6
FLOPS computation device bug fix (#1447) 3 years ago
  Glenn Jocher 9f27902368 Improved FLOPS computation (#1398) 3 years ago
  Glenn Jocher d8f5fcfe87
Improved FLOPS computation (#1398) 3 years ago
  Glenn Jocher fe341fa44d
Utils reorganization (#1392) 3 years ago
  Glenn Jocher 19e2482458
PyTorch version to screen and cleanup (#1325) 3 years ago
  Glenn Jocher c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates (#1233) 3 years ago
  Glenn Jocher 3b57cb5641
Simplified inference (#1153) 3 years ago
  Glenn Jocher 124f0e8212 torchvision nms bug fix 4 years ago
  Glenn Jocher 883924d9dc classifier, export, torch seed updates 4 years ago
  Glenn Jocher 89655a84f2 .fuse() gradient introduction bug fix 4 years ago
  tkianai c5969f79f5
add support for DWConv fuse operation (#898) 4 years ago
  Glenn Jocher b42e8a531b optimize imports 4 years ago
  Francisco Ingham 26c3b11fdc
Update torch_utils.py (#751) 4 years ago
  Glenn Jocher f5da528d28 reformat code 4 years ago
  NanoCode012 4949401a94
Fix redundant outputs via Logging in DDP training (#500) 4 years ago
  Glenn Jocher 3c6e2f7668
Single-source training (#680) 4 years ago
  Liu Changyu c020875b17
PyTorch 1.6.0 update with native AMP (#573) 4 years ago
  Glenn Jocher 1d17b9af0f
update yolo.py TTA flexibility and extensibility (#506) 4 years ago
  Glenn Jocher 9da56b62dd
v2.0 Release (#491) 4 years ago
  Glenn Jocher 2377e5f6ee FP16 EMA bug fix 4 years ago
  Glenn Jocher 41ff83a98d update torch_utils.py to FP16 EMA 4 years ago
  Glenn Jocher a586751904 multi-gpu ckpt filesize bug fix #253 4 years ago
  Glenn Jocher 24c5a941f0 --resume EMA fix #292 4 years ago
  Glenn Jocher 121d90b3f2 update fuse_conv_and_bn() 4 years ago
  Glenn Jocher 38f5c1ad1d pruning and sparsity initial commit 4 years ago
  NanoCode012 ccfa9373ae
Update to remove "module" attribute check 4 years ago
  NanoCode012 f83521ab9a
Fix save error for multi-gpu 4 years ago
  Glenn Jocher af41083d52 EMA FP16 fix #279 4 years ago
  Glenn Jocher df224a0d8f EMA bug fix #279 4 years ago
  Glenn Jocher f02481c73a
Update torch_utils.py 4 years ago
  Lornatang 8b8b792802 There is no need to download extra packages, official bring it with you 4 years ago
  Glenn Jocher 8fe299f179 model fuse 4 years ago
  Glenn Jocher b810b21299 augmented inference 4 years ago