35 Commits (412b113ac694d8fca9253668dad78f3550d81ea8)

Author SHA1 Message Date
  Glenn Jocher 9f27902368 Improved FLOPS computation (#1398) 4 years ago
  Glenn Jocher d8f5fcfe87
Improved FLOPS computation (#1398) 4 years ago
  Glenn Jocher fe341fa44d
Utils reorganization (#1392) 4 years ago
  Glenn Jocher 19e2482458
PyTorch version to screen and cleanup (#1325) 4 years ago
  Glenn Jocher c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates (#1233) 4 years ago
  Glenn Jocher 3b57cb5641
Simplified inference (#1153) 4 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
  Glenn Jocher c12023f145 updates 4 years ago
  Glenn Jocher 22d6088205 speed-reproducibility fix #17 4 years ago
  Glenn Jocher 1e84a23f38 initial commit 4 years ago