Glenn Jocher
57ef676af2
Update check_requirements() multiple string ( #9356 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-10 12:14:31 +03:00
Glenn Jocher
e9ddc5b527
Update check_requirements(args, cmds='') ( #9355 )
...
* Update `check_requirements(args, cmds='')`
* [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-10 12:00:16 +03:00
Glenn Jocher
24bf9cceb4
Update check_requirements() single install ( #9353 )
...
* Update `check_requirements()` single install
Faster install and better conflict resolution with single installation
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update
* Update
* Update
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-10 11:11:56 +03:00
Glenn Jocher
7aa263c5f2
Update DetectMultiBackend for tuple outputs 2 ( #9275 )
...
* Update DetectMultiBackend for tuple outputs 2
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update
* Update
* Update
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-09-04 15:39:57 +02:00
Glenn Jocher
5d4787baab
cudnn.benchmark = True on Seed 0 (#9259 )
...
* `cudnn.benchmark = True` on Seed 0
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* 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>
2022-09-02 16:05:23 +02:00
Glenn Jocher
ea98199041
Update general.py ( #9252 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-01 19:36:27 +02:00
Glenn Jocher
55b009616b
Python check warning emoji ( #9238 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-31 18:53:46 +02:00
Glenn Jocher
da22e01a68
check_font() fstring update (#9208 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-29 18:10:19 +02:00
Glenn Jocher
f65081c436
Add check_git_status(..., branch='master') argument ( #9199 )
...
Add check_git_status(branch='master') argument
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-29 15:00:55 +02:00
Glenn Jocher
e57275a2d8
Skip AMP check on MPS ( #9189 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-29 02:00:42 +02:00
Glenn Jocher
d07ddc6996
New TryExcept decorator ( #9154 )
...
* New TryExcept decorator
* [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-08-25 14:34:26 +02:00
Glenn Jocher
d0fa0042bd
New @try_export decorator ( #9096 )
...
* New export decorator
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* New export decorator
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* rename fcn to func
* rename to @try_export
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-23 13:06:33 +02:00
Glenn Jocher
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
Glenn Jocher
c725511bfc
Refactor for simplification ( #9054 )
...
* Refactor for simplification
* cleanup
* [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-08-21 01:34:03 +02:00
Glenn Jocher
d40cd0d454
Improved Profile() inference timing ( #9024 )
...
* Improved `Profile()` class
* Update predict.py
* Update val.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update val.py
* Update AutoShape
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-18 19:55:38 +02:00
Glenn Jocher
e83b422a69
Update dataset names from array to dictionary ( #9000 )
...
* Migrate dataset names to dictionary
* fix check
* backwards compat
* predict fix
* val fix
* Keep dataset stats behavior identical
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-17 17:52:53 +02:00
Glenn Jocher
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
Glenn Jocher
464c2c6713
Colab and Kaggle loggers removeHandler ( #8985 )
...
Resolve duplicate outputs in colab. Also optimizes Colab and Kaggle environment checks.
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-08-16 16:49:32 +02:00
Glenn Jocher
d7bc5d7773
Add .tar support for datasets ( #8963 )
2022-08-14 23:06:42 +02:00
Glenn Jocher
7639e4c7ef
Fix windows LOGGER with emojis output ( #8958 )
...
* Fix duplicate LOGGER output
* Update general.py
2022-08-14 00:12:09 +02:00
Kalen Michael
11a76bb4f5
Allow train to use zipped content ( #8940 )
...
* fix: added transparent image and empty alt to social bar
* fix: remove underlines
* fix: hide underlines for all social links
* fix: always unzip into folder
2022-08-13 03:39:09 +02:00
Code-Bat
b551098d59
About "os.R_OK" in general.py ( #8909 )
...
The notes says "Return True if directory has write permissions", however, the code below is "os.R_OK", I think "os.W_OK" is preferred.
2022-08-09 16:59:36 +02:00
Glenn Jocher
8f28ea4945
emoji-safe default logging ( #8888 )
...
* emoji-safe default logging
* implement everywhere
2022-08-06 12:30:08 +02:00
Victor Sonck
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
Glenn Jocher
39ce8ca19a
Remove assert not is_docker() from GitHub checks ( #8813 )
...
* Update
* Update
2022-08-01 03:01:44 +02:00
Amir Pourmand
afa403a809
Check git status on upstream ultralytics or origin dynamically ( #8694 )
...
* Add remote ultralytics and check git status with that
* Simplify
* Update general.py
* Update general.py
* s fix
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-07-31 18:17:38 +02:00
Max Strobel
0b5ac224ae
fix: broken `is_docker` check ( #8711 )
...
Checking if ``/workspace`` exists is not a reliable method to check if
the process runs in a docker container.
Reusing the logic from the npm "is-docker" package to check if the
process runs in a container.
References: https://github.com/sindresorhus/is-docker/blob/main/index.js
Fixes #8710 .
Co-authored-by: Maximilian Strobel <Maximilian.Strobel@infineon.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-07-26 19:02:44 +02:00
Glenn Jocher
7f7bd6fbcd
Set torch.cuda.manual_seed_all() for DDP ( #8688 )
...
* Set `torch.cuda.manual_seed_all()` for DDP
* Update general.py
* Update general.py
2022-07-23 18:24:14 +02:00
Glenn Jocher
b17629e54f
Update warning emojis ( #8678 )
2022-07-22 15:23:22 +02:00
Glenn Jocher
fbe67e4653
Fix OMP_NUM_THREADS=1 for macOS ( #8624 )
...
Resolves https://github.com/ultralytics/yolov5/issues/8623
2022-07-18 17:53:30 +02:00
Glenn Jocher
dd28df98c2
Avoid FP64 ops for MPS support in train.py ( #8511 )
...
Avoid FP64 ops for MPS support
Resolves https://github.com/ultralytics/yolov5/pull/7878#issuecomment-1177952614
2022-07-07 20:36:23 +02:00
Ayush Chaurasia
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
Glenn Jocher
a2a1ed201d
logging.ERROR on RANK not in (0, 1) (#8284 )
...
Improve DDP console response in https://github.com/ultralytics/yolov5/issues/8283
2022-06-21 17:41:14 +02:00
HRan2004
cf298fb72f
fix(utils): missing edge when resample segments ( #8092 )
...
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-06-09 10:52:05 +02:00
Glenn Jocher
47233e1698
Improved dataset error introspection ( #8091 )
...
* Improved dataset error introspection
Intended to help https://github.com/ultralytics/yolov5/issues/8090
* Update general.py
2022-06-03 12:45:51 +02:00
Glenn Jocher
5580b27daf
check_requirements() "" Windows double quote (#8016 )
2022-05-28 11:13:44 +02:00
Glenn Jocher
a6e99e4d66
cURL download double quotes ( #7999 )
...
Attempt to resolve Windows issues
2022-05-26 20:37:48 +02:00
Glenn Jocher
cd71aa3515
check_requirements() Windows fix (#7997 )
2022-05-26 19:01:46 +02:00
Glenn Jocher
d07f9ce0ea
AMP check improvements backup YOLOv5n pretrained ( #7959 )
...
* Reduce AMP check to detections verification
More robust and faster
* Update general.py
* Update general.py
2022-05-24 15:33:35 +02:00
Glenn Jocher
68ff6c9c08
Increase NMS fixed time limit 300ms + 30ms/img ( #7957 )
...
* Increase NMS fixed time limit to 200ms
* Update general.py
2022-05-24 14:11:03 +02:00
Glenn Jocher
b5ebd47d64
Increase NMS time limit to 50 ms/img ( #7956 )
2022-05-24 13:58:10 +02:00
Glenn Jocher
541a5b72bb
Windows check_file() fix ( #7938 )
...
Resolves Ultralytics HUB CI errors.
2022-05-23 16:15:38 +02:00
Glenn Jocher
9b4e05439c
Fix AMP check tolerance ( #7937 )
...
Adjust to 5%, fixes failing Colab AMP check with V100 (1.5% different) with 200% safety margin.
2022-05-23 15:57:16 +02:00
Glenn Jocher
cf3fb58522
AMP check image download backup ( #7936 )
...
Resolves https://github.com/ultralytics/yolov5/discussions/7931
2022-05-23 15:31:54 +02:00
Glenn Jocher
cee5959c74
Code refactor ( #7923 )
...
* Code refactor for general.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update restapi.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-05-22 15:24:08 +02:00
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
3356f2609c
Add Python version output ( #7814 )
2022-05-14 18:30:47 +02:00
Glenn Jocher
4a295b1a89
Add @threaded decorator ( #7813 )
...
* Add `@threaded` decorator
* [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-05-14 16:12:08 +02:00
Glenn Jocher
c4862fcd31
Update Dockerfile-cpu to force python3.9 ( #7675 )
...
* Update
* Update
* Do not install torch
* Pillow>9.1
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile-cpu
* Update Dockerfile
* fix emoji
* reduce scipy
* add libpython3.9
* Update Dockerfile
2022-05-02 20:53:02 -07:00
