Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

W&B: suppress warnings #4257

Merged
merged 33 commits into from
Aug 1, 2021
Merged

W&B: suppress warnings #4257

Changes from all commits
Commits
Show all changes
33 commits
Select commit Hold shift + click to select a range
d0d89b6
Improve docstrings and run names
AyushExel Jul 27, 2021
a283de6
default wandb login prompt with timeout
AyushExel Jul 27, 2021
33ab468
Merge branch 'ultralytics:master' into master
AyushExel Jul 27, 2021
78d1ed0
return key
AyushExel Jul 27, 2021
bba130b
Merge branch 'master' of https://github.com/AyushExel/yolov5
AyushExel Jul 27, 2021
725bd4a
Update api_key check logic
AyushExel Jul 27, 2021
bb372f5
Properly support zipped dataset feature
AyushExel Jul 28, 2021
27c06f0
update docstring
AyushExel Jul 28, 2021
3cafca4
merge master
AyushExel Jul 28, 2021
e64525a
Revert tuorial change
AyushExel Jul 28, 2021
0a67b4a
extend changes to log_dataset
AyushExel Jul 28, 2021
80eb6aa
add run name
AyushExel Jul 28, 2021
3b039be
bug fix
AyushExel Jul 28, 2021
31f04a2
bug fix
AyushExel Jul 28, 2021
2482fb3
Update comment
AyushExel Jul 28, 2021
6012e3f
fix import check
AyushExel Jul 28, 2021
66e36ef
remove unused import
AyushExel Jul 28, 2021
aa58d9e
Hardcore .yaml file extension
AyushExel Jul 28, 2021
7225c19
Merge branch 'ultralytics:master' into fix_dataset_check
AyushExel Jul 28, 2021
dfacf7b
reduce code
glenn-jocher Jul 28, 2021
3593b48
Merge branch 'ultralytics:master' into fix_dataset_check
AyushExel Jul 28, 2021
ab11132
Reformat using pycharm
AyushExel Jul 28, 2021
5d61242
Remove redundant try catch
AyushExel Jul 28, 2021
0a43ca7
More refactoring and bug fixes
AyushExel Jul 28, 2021
4830a54
merge master
AyushExel Jul 28, 2021
570381d
retry
AyushExel Jul 28, 2021
28d7cba
Reformat using pycharm
AyushExel Jul 28, 2021
c72f59f
respect LOGGERS include list
AyushExel Jul 28, 2021
23ef166
Merge branch 'fix_dataset_check' of https://github.com/AyushExel/yolo…
AyushExel Jul 28, 2021
e0e41c9
Merge branch 'ultralytics:master' into fix_dataset_check
AyushExel Jul 28, 2021
2b664dd
Merge branch 'ultralytics:master' into fix_dataset_check
AyushExel Jul 30, 2021
c6c5f3e
Merge branch 'ultralytics:master' into fix_dataset_check
AyushExel Aug 1, 2021
2dfbdda
call wandblogger.log instead of wandb.log
AyushExel Aug 1, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion utils/loggers/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,8 @@ def on_train_end(self, last, best, plots, epoch):
self.tb.add_image(f.stem, np.asarray(Image.open(f)), epoch, dataformats='HWC')

if self.wandb:
wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})
# Calling wandb.log. TODO: Refactor this into WandbLogger.log_model
wandb.log_artifact(str(best if best.exists() else last), type='model',
name='run_' + self.wandb.wandb_run.id + '_model',
aliases=['latest', 'best', 'stripped'])
Expand Down