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

Log command line options, hyperparameters, and weights per run in runs/ #104

Merged
merged 40 commits into from
Jul 9, 2020
Merged
Changes from 1 commit
Commits
Show all changes
40 commits
Select commit Hold shift + click to select a range
a85e6d0
add parser arg for hyp yaml file
alexstoken Jun 16, 2020
d9f446c
add save yaml of opt and hyp to tensorboard log_dir in train()
alexstoken Jun 16, 2020
4418809
change weights dir (wdir) to be unique to each run, under log_dir
alexstoken Jun 16, 2020
490f1e7
add save_dir arg to plot_lr_scheduler, default to current dir.
alexstoken Jun 16, 2020
25e51bc
add util function to get most recent last.pt file
alexstoken Jun 16, 2020
a448c3b
add logic for resuming and getting hyp for resume run
alexstoken Jun 16, 2020
333f678
add update default hyp dict with provided yaml
alexstoken Jun 16, 2020
5f2eeba
remove old print statements
alexstoken Jun 16, 2020
d342917
Fix get_latest_run() to search inside 'weights' subfolders
alexstoken Jun 17, 2020
3263a20
Fix get_latest_run to search 'search_dir' recursivly
alexstoken Jun 17, 2020
ade023c
Fix hyp file read in and dict update.
alexstoken Jun 17, 2020
3b2b330
Move results.txt from weights/ to log_dir
alexstoken Jun 17, 2020
945307b
Add save_dir to plot_lr_scheduler and plot_labels
alexstoken Jun 17, 2020
9b7386f
Add save_dir arg to test.test, use arg as location for saving batch jpgs
alexstoken Jun 17, 2020
c8152c8
Syntax fixes
alexstoken Jun 17, 2020
e572bb0
Add plot_results save location to log_dir
alexstoken Jun 21, 2020
ccf0af1
Revert coco128.yaml to initial commit
alexstoken Jun 24, 2020
d64ad0f
Remove --resume functionality and related checks/logic.
alexstoken Jun 24, 2020
7edbf65
Fix help message for cfg files
alexstoken Jun 24, 2020
7abf202
Mode all optimizer settings to 'hyp.yaml', integrate proper momentum …
alexstoken Jun 24, 2020
bc4ef48
Default optimizer SGD
alexstoken Jun 24, 2020
611aacf
Turn opt into dictionary before sending it to yaml
alexstoken Jun 24, 2020
d1ca6f2
Delete equip_hyp.yaml
alexstoken Jun 24, 2020
2d396be
Fix bug in --help from percent sign in help string
alexstoken Jun 24, 2020
de19165
Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp di…
alexstoken Jun 24, 2020
e18e681
Merge branch 'master' into advanced_logging
alexstoken Jun 27, 2020
f517ba8
Merge branch 'master' into advanced_logging
alexstoken Jul 6, 2020
5ac517b
Remove duplicate `verbose` arg in test.py
alexstoken Jul 7, 2020
8b6dbb7
Add optimizer choice to hyp file
alexstoken Jul 7, 2020
52bac22
Add in --resume functionality with option to specify path or to get m…
alexstoken Jul 7, 2020
95f0a56
Bug fix to get_latest_run when recent run is named
alexstoken Jul 7, 2020
4364751
Colab example/test of new features
alexstoken Jul 7, 2020
e6e7e7f
Delete advanced_logging_test.ipynb
alexstoken Jul 7, 2020
c654d18
Update utils.py
glenn-jocher Jul 8, 2020
cbe39a1
Update utils.py
glenn-jocher Jul 8, 2020
c3d3e6b
Update test.py
glenn-jocher Jul 8, 2020
22ab1c2
Update test.py
glenn-jocher Jul 8, 2020
6b134d9
Update train.py
glenn-jocher Jul 8, 2020
abb024d
Delete new_hyp.yaml
glenn-jocher Jul 8, 2020
dc5e183
Merge branch 'master' into advanced_logging
glenn-jocher Jul 9, 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
Prev Previous commit
Next Next commit
Turn opt into dictionary before sending it to yaml
alexstoken committed Jun 24, 2020

Verified

This commit was signed with the committer’s verified signature.
renovate-bot Mend Renovate
commit 611aacf1bfc63a6bb556a5f27ac5e560cd8b1191
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
@@ -193,7 +193,7 @@ def train(hyp):
yaml.dump(hyp, f)

with open(os.path.join(log_dir, 'opt.yaml'), 'w') as f:
yaml.dump(opt, f)
yaml.dump(vars(opt), f)

# Class frequency
labels = np.concatenate(dataset.labels, 0)