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

Weights & Biases (W&B) Feature Addition #1235

Merged
merged 18 commits into from
Oct 31, 2020
Merged

Weights & Biases (W&B) Feature Addition #1235

Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
60d6a4b
Add wandb metric logging and bounding box debugging
Oct 7, 2020
15779b5
Improve formatting, readability
Oct 8, 2020
ee84468
Remove mutliple path for init, improve formatting
AyushExel Oct 8, 2020
b534855
Add wandb params
Oct 28, 2020
6fe4957
Remove typecasting in bbox coordinates and reformat
Oct 28, 2020
40b83a3
Cleanup
Oct 28, 2020
be999ec
add wandb to requirements.txt
glenn-jocher Oct 31, 2020
5e143be
minor updates to test.py
glenn-jocher Oct 31, 2020
ce58596
general reorg
glenn-jocher Oct 31, 2020
82a95d4
reduce --log-imgs to 10
glenn-jocher Oct 31, 2020
7aa485a
clean wandb import
glenn-jocher Oct 31, 2020
a964353
reverse wandb import assert
glenn-jocher Oct 31, 2020
c282a2b
add except AssertionError to try import
glenn-jocher Oct 31, 2020
5a7e009
move wandb init to all global ranks
glenn-jocher Oct 31, 2020
9327313
replace print() with logger.info()
glenn-jocher Oct 31, 2020
1d5570d
replace print() with logger.info()
glenn-jocher Oct 31, 2020
da3edca
move wandb.init() bug fix
glenn-jocher Oct 31, 2020
caff798
project PosixPath to basename bug fix
glenn-jocher Oct 31, 2020
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
9 changes: 5 additions & 4 deletions train.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -118,6 +118,11 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf)
# plot_lr_scheduler(optimizer, scheduler, epochs)

# Logging
if wandb and wandb.run is None:
id = ckpt.get('wandb_id') if 'ckpt' in locals() else None
wandb_run = wandb.init(config=opt, resume="allow", project=log_dir, id=id)

glenn-jocher marked this conversation as resolved.
Show resolved Hide resolved
# Resume
start_epoch, best_fitness = 0, 0.0
if pretrained:
Expand All @@ -141,10 +146,6 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
(weights, ckpt['epoch'], epochs))
epochs += ckpt['epoch'] # finetune additional epochs

# Logging
if wandb and wandb.run is None:
wandb_run = wandb.init(config=opt, resume="allow", project=log_dir, id=ckpt.get('wandb_id'))

del ckpt, state_dict

# Image sizes
Expand Down