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.

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

Improved W&B integration #2125

Merged
merged 84 commits into from
Mar 22, 2021
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
84 commits
Select commit Hold shift + click to select a range
ba39bfd
Init Commit
AyushExel Feb 2, 2021
5fcd1dc
new wandb integration
AyushExel Feb 3, 2021
c540e3b
Update
AyushExel Feb 3, 2021
8253b24
Use data_dict in test
AyushExel Feb 3, 2021
7f89535
Updates
AyushExel Feb 3, 2021
c149930
Update: scope of log_img
AyushExel Feb 3, 2021
49edb90
Update: scope of log_img
AyushExel Feb 3, 2021
7922683
Update
AyushExel Feb 3, 2021
e1e7179
Update: Fix logging conditions
AyushExel Feb 3, 2021
e632514
Add tqdm bar, support for .txt dataset format
AyushExel Feb 9, 2021
3e8f4ae
Improve Result table Logger
AyushExel Feb 21, 2021
cd094f3
Init Commit
AyushExel Feb 2, 2021
aa5231e
new wandb integration
AyushExel Feb 3, 2021
0fdf3d3
Update
AyushExel Feb 3, 2021
37a2ed6
Use data_dict in test
AyushExel Feb 3, 2021
ac7d4b1
Updates
AyushExel Feb 3, 2021
ebc1d18
Update: scope of log_img
AyushExel Feb 3, 2021
745a272
Update: scope of log_img
AyushExel Feb 3, 2021
7679454
Update
AyushExel Feb 3, 2021
b8210a7
Update: Fix logging conditions
AyushExel Feb 3, 2021
ac9a613
Add tqdm bar, support for .txt dataset format
AyushExel Feb 9, 2021
4f7c150
Improve Result table Logger
AyushExel Feb 21, 2021
b8bbfce
Merge branch 'wandb_clean' of https://github.com/AyushExel/yolov5 int…
AyushExel Feb 23, 2021
c1e6697
Add dataset creation in training script
AyushExel Feb 23, 2021
1948562
Change scope: self.wandb_run
AyushExel Feb 23, 2021
8848f3c
Add wandb-artifact:// natively
AyushExel Feb 25, 2021
deca116
Add suuport for logging dataset while training
AyushExel Feb 26, 2021
20185f2
Cleanup
AyushExel Feb 26, 2021
5287a79
Merge branch 'master' into wandb_clean
AyushExel Feb 26, 2021
e13994d
Fix: Merge conflict
AyushExel Feb 26, 2021
1080952
Fix: CI tests
AyushExel Feb 26, 2021
5a859d4
Automatically use wandb config
AyushExel Feb 27, 2021
519cb7d
Fix: Resume
AyushExel Feb 28, 2021
3242f52
Fix: CI
AyushExel Feb 28, 2021
8128216
Enhance: Using val_table
AyushExel Feb 28, 2021
043befa
More resume enhancement
AyushExel Feb 28, 2021
c2d98f0
FIX : CI
AyushExel Feb 28, 2021
dbb69f4
Add alias
AyushExel Feb 28, 2021
8505a58
Get useful opt config data
AyushExel Mar 1, 2021
04f8880
train.py cleanup
AyushExel Mar 2, 2021
27a33dd
Merge remote-tracking branch 'upstream/master' into wandb_clean
AyushExel Mar 2, 2021
54dee24
Cleanup train.py
AyushExel Mar 2, 2021
21a15a5
more cleanup
AyushExel Mar 2, 2021
d38c620
Cleanup| CI fix
AyushExel Mar 2, 2021
e5400ba
Reformat using PEP8
AyushExel Mar 3, 2021
45e2c55
FIX:CI
AyushExel Mar 3, 2021
75f31d0
Merge remote-tracking branch 'upstream/master' into wandb_clean
AyushExel Mar 6, 2021
613b102
rebase
AyushExel Mar 6, 2021
9772645
remove uneccesary changes
AyushExel Mar 6, 2021
cd1237e
remove uneccesary changes
AyushExel Mar 6, 2021
d172ba1
remove uneccesary changes
AyushExel Mar 6, 2021
7af0186
remove unecessary chage from test.py
AyushExel Mar 6, 2021
51dca6d
FIX: resume from local checkpoint
AyushExel Mar 8, 2021
1438483
FIX:resume
AyushExel Mar 8, 2021
e7d18c6
FIX:resume
AyushExel Mar 8, 2021
22d97a7
Reformat
AyushExel Mar 8, 2021
8e97cdf
Performance improvement
AyushExel Mar 9, 2021
2ffb643
Fix local resume
AyushExel Mar 9, 2021
7836d17
Fix local resume
AyushExel Mar 9, 2021
aa785ec
FIX:CI
AyushExel Mar 9, 2021
f97446e
Fix: CI
AyushExel Mar 9, 2021
807a0e1
Imporve image logging
AyushExel Mar 9, 2021
20b4450
(:(:Redo CI tests:):)
AyushExel Mar 9, 2021
db81c64
Remember epochs when resuming
AyushExel Mar 9, 2021
25ff6b8
Remember epochs when resuming
AyushExel Mar 9, 2021
819ebec
Update DDP location
glenn-jocher Mar 10, 2021
b23a902
merge master
glenn-jocher Mar 14, 2021
f742857
PEP8 reformat
glenn-jocher Mar 14, 2021
350b8ab
0.25 confidence threshold
glenn-jocher Mar 14, 2021
395379e
reset train.py plots syntax to previous
glenn-jocher Mar 14, 2021
a06b25c
reset epochs completed syntax to previous
glenn-jocher Mar 14, 2021
cc49f6a
reset space to previous
glenn-jocher Mar 14, 2021
2d56697
remove brackets
glenn-jocher Mar 14, 2021
ba859a6
reset comment to previous
glenn-jocher Mar 14, 2021
52e3e71
Update: is_coco check, remove unused code
AyushExel Mar 14, 2021
ad1ad8f
Remove redundant print statement
AyushExel Mar 14, 2021
72dd23b
Remove wandb imports
AyushExel Mar 14, 2021
ac955ab
remove dsviz logger from test.py
AyushExel Mar 14, 2021
8bded54
Remove redundant change from test.py
AyushExel Mar 14, 2021
1aca390
remove redundant changes from train.py
AyushExel Mar 14, 2021
4c1c9bf
reformat and improvements
AyushExel Mar 20, 2021
f4923b4
Fix typo
AyushExel Mar 21, 2021
af23506
Merge branch 'master' of https://github.com/ultralytics/yolov5 into w…
AyushExel Mar 21, 2021
ca06d31
Add tqdm tqdm progress when scanning files, naming improvements
AyushExel Mar 21, 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
Prev Previous commit
Next Next commit
Updates
  • Loading branch information
AyushExel committed Feb 23, 2021
commit ac7d4b1533be210349bc2dbc339937fa377d543f
1 change: 0 additions & 1 deletion utils/wandb_logging/__init__.py
Original file line number Diff line number Diff line change
@@ -1 +0,0 @@

1 change: 0 additions & 1 deletion utils/wandb_logging/wandb_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,6 @@ def download_model_artifact(self, name):
return modeldir, model_artifact

def log_model(self, path, opt, epoch):
print("Logging model, ", epoch)
datetime_suffix = datetime.today().strftime('%Y-%m-%d-%H-%M-%S')
model_artifact = wandb.Artifact('run_' + wandb.run.id + '_model', type='model', metadata={
'original_url': str(path),
Expand Down