25fb83f287
Revert "work with later versions of pytorch"
...
This reverts commit 19c93d5af4
2025-02-03 10:02:13 +11:00
19c93d5af4
work with later versions of pytorch
2025-01-24 19:47:09 +11:00
541ae60ee9
trim logger code
2024-12-28 22:33:09 +11:00
81f81d3e97
Fixed randrange warnings in Python>=3.10 ( #11161 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-03-23 20:30:10 +01:00
7a972e86c4
Update .pre-commit-config.yaml ( #11009 )
...
* Update .pre-commit-config.yaml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update __init__.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update .pre-commit-config.yaml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Pre-commit updates
* Pre-commit updates
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-18 01:06:24 +01:00
a2de5c5bf6
Subprocess improvements ( #10973 )
...
* Use list-form arguments for subprocess.run calls where possible
Augments #10944
* Deduplicate curl code
* Avoid eval() to parse integer
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-13 18:00:31 +04:00
238da321cb
Security3 ( #10944 )
...
* Security improvements
* Security improvements
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
---------
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-09 17:58:24 +04:00
10e93d295f
Set a seed of generator with an option for more randomness when training several models with different seeds ( #10486 )
...
* set seed with parameter
Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
* make seed to be a large number
* set seed with a parameter
* set a seed of dataloader with opt for more randomness
Signed-off-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-12-19 10:27:34 +01:00
40bb8030f8
Scope gitpyhon import in `check_git_info()` ( #10221 )
...
* Scope gitpyhon import in `check_git_info()`
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 14:22:07 +01:00
0307954e4e
Add git info to cls, seg checkpoints ( #10217 )
2022-11-19 03:32:21 +01:00
9286336cb4
Add `git` info to training checkpoints ( #9655 )
...
* Add git status on train checkpoints
* Update
* Update
* Update
* Update
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update general.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-19 03:16:17 +01:00
0322bb3196
New global `TQDM_BAR_FORMAT` ( #10211 )
...
* New global TQDM_BAR_FORMAT
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-18 21:39:13 +01:00
fde7758468
YOLOv5 AutoCache Update ( #10027 )
...
* AutoCache
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* AutoCache
* AutoCache
* AutoCache
* AutoCache
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-04 15:30:12 +01:00
02b8a4c21b
Update default `--epochs 100` ( #10024 )
...
* Update default `--epochs 100`
@AyushExel @kalenmike updating default Detection and Segmentation trainings to 100 epochs
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update train.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README_cn.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-11-03 17:58:15 +01:00
acff977af3
Logger Cleanup ( #9828 )
2022-10-17 15:30:42 +02:00
209be932de
Fix segment evolution keys ( #9742 )
...
* Update
* Cleanup
2022-10-09 23:51:29 +02:00
ca9c993d6c
Standardize warnings with `WARNING ⚠️ ...` ( #9467 )
...
* Standardize warnings
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-18 16:15:25 +02:00
3cd66b1c38
Train.py line 486 typo fix ( #9330 )
...
fixed issue
Signed-off-by: robinned <78896580+robinned@users.noreply.github.com>
Signed-off-by: robinned <78896580+robinned@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2022-09-08 22:00:54 +03:00
903b239f13
Comet Logging and Visualization Integration ( #9232 )
...
* add comet to logger interface
* add comet logger
* add support for updated parameters
* clean up offline logger creation
* update callback args for comet logger
* add comet optimizer
* add optimizer config
* add comet README
* update tutorial notebook with Comet section
* add option to log class level metrics
* add support for class level metrics and confusion matrix
* handle errors when adding files to artifacts
* fix typo
* clean resume workflow
* updates for HPO
* update comet README
* fix typo in comet README
* update code snippets in comet README
* update comet links in tutorial
* updated links
* change optimizer batch size param and update comet README image
* update comet section in tutorial
* use prexisting cmd line flags to configure logger
* update artifact upload/download flow
* remove come remove comet logger specific cmd line args
* move downloading weights into comet logger code
* remove extra argparse
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* change checkpoint logging flow to follow offline logger
* update resume flow
* add comet logger to remote dataset property
* update cmd line args in hpo
* set types for integer/float env variables
* update README
* fix typo in README
* default to always logging model predictions
* Update tutorial.ipynb
* Update train.py
* Add Comet to Integrations table
* Update README.md
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-07 18:28:46 +03:00
2d082a07bd
Refactor Loggers : Move code outside train.py ( #9241 )
...
* update
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* update
* update
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-01 19:17:36 +02:00
06831aa9e9
Improved Usage example docstrings ( #9075 )
...
* Updated Usage examples
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update predict.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-22 01:06:29 +02:00
8665d557c1
Threaded TensorBoard graph logging ( #9070 )
...
* Log TensorBoard graph on pretrain_routine_end
* fix
2022-08-21 16:51:50 +02:00
0b8639a40a
Rename 'labels' to 'instances' ( #9066 )
...
* Rename labels to instances
* Rename labels to instances
* align val
2022-08-21 15:50:02 +02:00
e0700cce77
Daemon `plot_labels()` for faster start ( #9057 )
...
* Daemon `plot_labels()` for faster start
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update train.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-21 03:47:37 +02:00
fdcb92a938
Update train.py `import val as validate` ( #9037 )
...
* Update train.py `import val as validate`
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-19 16:54:58 +02:00
781401ec70
Default --data `imagenette160` training (fastest) ( #9033 )
...
* Default --data `imagenette160` training (fastest)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update train.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update train.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update train.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-19 15:06:37 +02:00
20049be2e7
EMA FP32 assert classification bug fix ( #9016 )
...
* Return EMA float on classification val
* verbose val fix
* EMA check
2022-08-18 14:06:15 +02:00
d3ea0df8b9
New YOLOv5 Classification Models ( #8956 )
...
* Update
* Logger step fix: Increment step with epochs (#8654 )
* enhance
* revert
* allow training from scratch
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update --img argument from train.py
single line
* fix image size from 640 to 128
* suport custom dataloader and augmentation
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* format
* Update dataloaders.py
* Single line return, single line comment, remove unused argument
* address PR comments
* fix spelling
* don't augment eval set
* use fstring
* update augmentations.py
* new maning convention for transforms
* reverse if statement, inline ops
* reverse if statement, inline ops
* updates
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* update dataloaders
* Remove additional if statement
* Remove is_train as redundant
* Cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup2
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update classifier.py
* Update augmentations.py
* fix: imshow clip warning
* update
* Revert ToTensorV2 removal
* Update classifier.py
* Update normalize values, revert uint8
* normalize image using cv2
* remove dedundant comment
* Update classifier.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* replace print with logger
* commit steps
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* 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
* Allow logging models from GenericLogger (#8676 )
* enhance
* revert
* allow training from scratch
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update --img argument from train.py
single line
* fix image size from 640 to 128
* suport custom dataloader and augmentation
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* format
* Update dataloaders.py
* Single line return, single line comment, remove unused argument
* address PR comments
* fix spelling
* don't augment eval set
* use fstring
* update augmentations.py
* new maning convention for transforms
* reverse if statement, inline ops
* reverse if statement, inline ops
* updates
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* update dataloaders
* Remove additional if statement
* Remove is_train as redundant
* Cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup2
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update classifier.py
* Update augmentations.py
* fix: imshow clip warning
* update
* Revert ToTensorV2 removal
* Update classifier.py
* Update normalize values, revert uint8
* normalize image using cv2
* remove dedundant comment
* Update classifier.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* replace print with logger
* commit steps
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* support final model logging
* update
* update
* update
* update
* remove curses
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update classifier.py
* Update __init__.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update
* Update
* Update
* Update
* Update dataset download
* Update dataset download
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Pass imgsz to classify_transforms()
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Cos scheduler
* Cos scheduler
* Remove unused args
* Update
* Add seed
* Add seed
* Update
* Update
* Add run(), main()
* Merge master
* Merge master
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Create YOLOv5 BaseModel class (#8829 )
* Create BaseModel
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* fix
* Hub load device fix
* Update
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
* Add experiment
* Merge master
* Attach names
* weight decay = 1e-4
* weight decay = 5e-5
* update smart_optimizer console printout
* fashion-mnist fix
* Merge master
* Update Table
* Update Table
* Remove destroy process group
* add kwargs to forward()
* fuse fix for resnet50
* nc, names fix for resnet50
* nc, names fix for resnet50
* ONNX CPU inference fix
* revert
* cuda
* if augment or visualize
* if augment or visualize
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* New smart_inference_mode()
* Update README
* Refactor into /classify dir
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* reset defaults
* reset defaults
* fix gpu predict
* warmup
* ema half fix
* spacing
* remove data
* remove cache
* remove denormalize
* save run settings
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* verbose false on initial plots
* new save_yaml() function
* Update ci-testing.yml
* Path(data) CI fix
* Separate classification CI
* fix val
* fix val
* fix val
* smartCrossEntropyLoss
* skip validation on hub load
* autodownload with working dir root
* str(data)
* Dataset usage example
* im_show normalize
* im_show normalize
* add imagenet simple names to multibackend
* Add validation speeds
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* 24-space names
* Update bash scripts
* Update permissions
* Add bash script arguments
* remove verbose
* TRT data fix
* names generator fix
* optimize if names
* update usage
* Add local loading
* Verbose=False
* update names printing
* Add Usage examples
* Add Usage examples
* Add Usage examples
* Add Usage examples
* named_children
* reshape_classifier_outputs
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* update
* update
* fix CI
* fix incorrect class substitution
* fix incorrect class substitution
* remove denormalize
* ravel fix
* cleanup
* update opt file printing
* update opt file printing
* update defaults
* add opt to checkpoint
* Add warning
* Add comment
* plot half bug fix
* Use NotImplementedError
* fix export shape report
* Fix TRT load
* cleanup CI
* profile comment
* CI fix
* Add cls models
* avoid inplace error
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Fix usage examples
* Update README
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
* Update README
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-17 11:59:01 +02:00
fd004f5648
Fix bug with resume ( #8912 )
...
* Fix bug with resume
* restore del on resume=false
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-17 00:58:30 +02:00
f1214f237d
Remove DDP `destroy_process_group()` on train end ( #8935 )
...
May resolve https://github.com/ultralytics/yolov5/issues/7307
2022-08-13 03:57:31 +02:00
c962db2749
Fix anchor incorrectly initialized ( #8891 )
...
* Fix anchor incorrectly initialized
Using --noautoanchor and --evolve simultaneously leads to anchor incorrectly initialized. --noautoanchor denotes anchors don't need to evolve, thus removing anchors from hyp will fix it.
* Update train.py
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-08 17:15:18 +02:00
20f1b7ea08
`smart_optimizer()` improved reporting ( #8887 )
...
Update smart_optimizer() weight_decay reporting
2022-08-06 11:30:54 +02:00
378bde4bba
ClearML experiment tracking integration ( #8620 )
...
* Add titles to matplotlib plots
* Add ClearML Experiment Tracking integration.
* Add ClearML Data Version Management automatic download when requested
* Add ClearML Hyperparameter Optimization
* ClearML save period integration
* Fix wandb breaking when used with ClearML dataset
* Fix wandb breaking when used with ClearML resume and dataset
* Add ClearML documentation
* fixed small bug in clearml integration that misreports epoch number
* Final ClearMl additions before refactor
* Add correct epoch reporting
* Add remote execution and autoscaling docs for ClearML integration
* Added images to clearml integration docs
* fixed logo alignment bug and added hpo screenshot clearml
* Fixed small epoch number bug in clearml integration
* Remove saved model flush clearml
* Cleanup clearml readme section
* Cleaned up clearml logger docstring
* Remove resume readme section clearml
* Clearml integration cleanup
* Updated ClearML documentation
* Added dark vs light icons ClearML Readme
* Clearml Readme styling
* Add better gifs
* Fixed gif file size
* Add better images in tutorial notebook
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Addressed comments in PR #8620
* Fixed circular import
* Fixed circular import
* Update tutorial.ipynb
* Update tutorial.ipynb
* Inline comment
* Restructured tutorial notebook
* Add correct ClearML link to README
* Update tutorial.ipynb
* Update general.py
* Update __init__.py
* Update __init__.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update __init__.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update __init__.py
* Update README.md
* Update __init__.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* spelling
* Update tutorial.ipynb
* notebook cutt.ly links
* Update README.md
* Update README.md
* cutt.ly links in tutorial
* Removed labels as they show up on last subplot only
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-05 20:50:49 +02:00
daed7a844e
`--resume` training from URL weights fix ( #8884 )
...
--resume training from URL weights fix
@kalenmike should fix data error on HUB resume
2022-08-05 16:42:10 +02:00
e073658e11
`--resume` training from URL weights ( #8882 )
...
@kalenmike
2022-08-05 16:27:28 +02:00
a75a1105a1
Self-contained checkpoint `--resume` ( #8839 )
...
* Single checkpoint resume
* Update train.py
* Add hyp
* Add hyp
* Add hyp
* FIX
* avoid resume on url data
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* avoid resume on url data
* avoid resume on url data
* Update
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-03 21:28:22 +02:00
08c8c3e00a
New `smart_resume()` ( #8838 )
...
* New `smart_resume()`
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* fix
2022-08-02 15:13:58 +02:00
f3c78a387e
Remove hook `torch.nan_to_num(x)` ( #8826 )
...
* Remove hook `torch.nan_to_num(x)`
Observed erratic training behavior (green line) with the nan_to_num hook in classifier branch. I'm going to remove it from master.
* Update train.py
2022-08-01 21:39:04 +02:00
0669f1b27b
Add tensor hooks and 10.0 gradient clipping ( #8598 )
...
* Add tensor hooks and gradient clipping https://github.com/ultralytics/yolov5/issues/8578
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Remove retain_grad(), because its not necessary
* Update train.py
* Simplify
* Update train.py
* Update train.py
* Update train.py
* Update train.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-01 12:08:46 +02:00
59595c1365
Update train.py for `val.run(half=amp)` ( #8804 )
...
Disable FP16 validation if AMP checks fail or amp=False.
2022-07-31 04:17:39 +02:00
9cf5fd5ac3
assert torch!=1.12.0 for DDP training ( #8621 )
...
* assert torch!=1.12.0 for DDP training
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-18 15:05:58 +02:00
51fb467b63
Refactor optimizer initialization ( #8607 )
...
* Refactor optimizer initialization
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update train.py
* Update train.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-17 11:43:52 +02:00
7204c1ca25
Explicitly set `weight_decay` value ( #8592 )
...
* explicitly set weight_decay value
The default weight_decay value of AdamW is 1e-2, so we should set it to zero.
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-07-16 15:51:48 +02:00
27d831b6e4
Training reproducibility improvements ( #8213 )
...
* attempt at reproducibility
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* use deterministic algs
* fix everything :)
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* revert dataloader changes
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* process_batch as np
* remove newline
* Remove dataloader init fcn
* Update val.py
* Update train.py
* revert additional changes
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update train.py
* Add --seed arg
* Update general.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update train.py
* Update train.py
* Update val.py
* Update train.py
* Update general.py
* Update general.py
* Add deterministic argument to init_seeds()
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-07-07 14:39:29 +02:00
e50dc38d36
Improve `--local_rank` arg comment ( #8409 )
...
* add more docs
* add more docs
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update train.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-06-30 15:01:31 +02:00
6935a54e60
Implementation of Early Stopping for DDP training ( #8345 )
...
* Implementation of Early Stopping for DDP training
This edit correctly uses the broadcast_object_list() function to send slave processes a boolean so to end the training phase if the variable is True, thus allowing the master process to destroy the process group and terminate.
* Update train.py
* Update train.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update train.py
* Update train.py
* Update train.py
* Further cleanup
This cleans up the definition of broadcast_list and removes the requirement for clear() afterward.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-06-29 12:41:46 +02:00
8ebf569d14
Fix bias warmup LR init ( #8356 )
...
Per https://github.com/ultralytics/yolov5/issues/8352
2022-06-27 16:11:24 +02:00
dcf8073881
Simplify and fix `--save-period` epoch 0 ( #8042 )
...
Simplify and update `--save-period` epoch 0
2022-05-30 12:42:12 +02:00
eb1217f3ac
Add PyTorch AMP check ( #7917 )
...
* Add PyTorch AMP check
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* Cleanup
* Cleanup
* Robust for DDP
* Fixes
* Add amp enabled boolean to check_train_batch_size
* Simplify
* space to prefix
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-05-22 13:41:18 +02:00
f00071416f
Refactor collections and fstrings ( #7821 )
...
* Update torch_utils.py
* Additional code refactoring
* tuples to sets
* Cleanup
2022-05-15 16:38:26 +02:00
Edward Yang
Edward Yang
imyhxy
Glenn Jocher
Aarni Koskela
Yonghye Kwon
robinned
Dhruv Nair
Ayush Chaurasia
Yana
黄不盈
Victor Sonck
UnglvKitDe
Amir Pourmand
Giacomo Guiduzzi