Glenn Jocher
0ff5aeca61
Create date_modified() ( #2616 )
Updated device selection string with fallback for non-git directories.
```python
def select_device(device='', batch_size=None):
# device = 'cpu' or '0' or '0,1,2,3'
s = f'YOLOv5 🚀 {git_describe() or date_modified()} torch {torch.__version__} ' # string
...
```
3年前
Glenn Jocher
77415a42e5
Update git_describe() for remote dir usage ( #2606 )
3年前
Glenn Jocher
9b11f0c58b
PyTorch Hub models default to CUDA:0 if available ( #2472 )
* PyTorch Hub models default to CUDA:0 if available
* device as string bug fix
3年前
Glenn Jocher
f01f3223d5
Integer printout ( #2450 )
* Integer printout
* test.py 'Labels'
* Update train.py
3年前
xiaowo1996
83dc1b4484
FLOPS min stride 32 ( #2276 )
Signed-off-by: xiaowo1996 <429740343@qq.com>
3年前
Glenn Jocher
ffef77124e
Update torch_utils.py ( #1895 )
3年前
Glenn Jocher
6ab589583c
Add colorstr() ( #1887 )
* Add colorful()
* update
* newline fix
* add git description
* --always
* update loss scaling
* update loss scaling 2
* rename to colorstr()
3年前
Glenn Jocher
69be8e738f
YOLOv5 v4.0 Release ( #1837 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update study.png
* update study.png
* Update datasets.py
3年前
Glenn Jocher
9f5a18bb80
Torch CUDA synchronize update ( #1826 )
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* newline
3年前
Glenn Jocher
c0ffcdf998
Display correct CUDA devices ( #1776 )
* Display correct CUDA devices
* cleanup
3年前
Glenn Jocher
3004fb5bc1
Automatic m.half() profile on x.half()
3年前
Glenn Jocher
0bd9c48609
Update torch_utils.py
FLOPS to GFLOPS
3年前
Glenn Jocher
394d1c89f3
Input channel yaml['ch'] addition ( #1741 )
3年前
Glenn Jocher
8bc0027afc
Update loss criteria constructor ( #1711 )
3年前
NanoCode012
035ac82ed0
Fix torch multi-GPU --device error ( #1701 )
* Fix torch GPU error
* Update torch_utils.py
single-line device =
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3年前
Glenn Jocher
ada90e3901
Profile() feature addition ( #1673 )
* Profile() feature addition
* cleanup
3年前
Glenn Jocher
8918e63476
Increase FLOPS robustness ( #1608 )
3年前
yujun
05a955a3f6
FLOPS computation device bug fix ( #1447 )
* Update torch_utils.py
fix issue#113 , inputs device should be same with model parameters' device
* Update torch_utils.py
* Update torch_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4年前
Glenn Jocher
9f27902368
Improved FLOPS computation ( #1398 )
4年前
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
* Improved FLOPS computation
* update comment
4年前
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4年前
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
4年前
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4年前
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4年前
Glenn Jocher
124f0e8212
torchvision nms bug fix
4年前
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
4年前
Glenn Jocher
89655a84f2
.fuse() gradient introduction bug fix
4年前
tkianai
c5969f79f5
add support for DWConv fuse operation ( #898 )
* the value should be a list to unpacking
* filter the None labels
* update readme to remove apex dependency
* add support for DWConv fuse operation
4年前
Glenn Jocher
b42e8a531b
optimize imports
4年前
Francisco Ingham
26c3b11fdc
Update torch_utils.py ( #751 )
4年前
Glenn Jocher
f5da528d28
reformat code
4年前
NanoCode012
4949401a94
Fix redundant outputs via Logging in DDP training ( #500 )
* Change print to logging
* Clean function set_logging
* Add line spacing
* Change leftover prints to log
* Fix scanning labels output
* Fix rank naming
* Change leftover print to logging
* Reorganized DDP variables
* Fix type error
* Make quotes consistent
* Fix spelling
* Clean function call
* Add line spacing
* Update datasets.py
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4年前
Glenn Jocher
3c6e2f7668
Single-source training ( #680 )
* Single-source training
* Extract hyperparameters into seperate files
* weight decay scientific notation yaml reader bug fix
* remove import glob
* intersect_dicts() implementation
* 'or' bug fix
* .to(device) bug fix
4年前
Liu Changyu
c020875b17
PyTorch 1.6.0 update with native AMP ( #573 )
* PyTorch have Automatic Mixed Precision (AMP) Training.
* Fixed the problem of inconsistent code length indentation
* Fixed the problem of inconsistent code length indentation
* Mixed precision training is turned on by default
4年前
Glenn Jocher
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
4年前
Glenn Jocher
9da56b62dd
v2.0 Release ( #491 )
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4年前
Glenn Jocher
2377e5f6ee
FP16 EMA bug fix
4年前
Glenn Jocher
41ff83a98d
update torch_utils.py to FP16 EMA
4年前
Glenn Jocher
a586751904
multi-gpu ckpt filesize bug fix #253
4年前
Glenn Jocher
24c5a941f0
--resume EMA fix #292
4年前
Glenn Jocher
121d90b3f2
update fuse_conv_and_bn()
4年前
Glenn Jocher
38f5c1ad1d
pruning and sparsity initial commit
4年前
NanoCode012
ccfa9373ae
Update to remove "module" attribute check
4年前
NanoCode012
f83521ab9a
Fix save error for multi-gpu
4年前
Glenn Jocher
af41083d52
EMA FP16 fix #279
4年前
Glenn Jocher
df224a0d8f
EMA bug fix #279
4年前
Glenn Jocher
f02481c73a
Update torch_utils.py
4年前
Lornatang
8b8b792802
There is no need to download extra packages, official bring it with you
I submitted it once in your yolov3 project, you seem to accept it? I'm not sure. I'll submit PR again.
4年前
Glenn Jocher
8fe299f179
model fuse
4年前
Glenn Jocher
b810b21299
augmented inference
4年前