Commit Graph
34 Commits
Author SHA1 Message Date
Alex Stoken 9d631408a2 Move hyp and opt yaml save to top of train()
Fixes bug where scaled values were saved in hyp.yaml, which would cause continuity issues with --resume
2020-07-09 16:18:55 -05:00
Alex Stoken e6e7e7f1ac Delete advanced_logging_test.ipynb
Remove from advanced_logging branch before merge.
2020-07-07 11:31:25 -05:00
Alex Stoken 43647514da Colab example/test of new features 2020-07-07 11:24:11 -05:00
Alex Stoken 95f0a56df7 Bug fix to get_latest_run when recent run is named 2020-07-07 11:19:49 -05:00
Alex Stoken 52bac22f09 Add in --resume functionality with option to specify path or to get most recent run 2020-07-07 10:42:28 -05:00
Alex Stoken 8b6dbb7cfc Add optimizer choice to hyp file 2020-07-07 10:30:48 -05:00
Alex Stoken 5ac517b22a Remove duplicate verbose arg in test.py 2020-07-07 10:09:53 -05:00
Alex Stoken f517ba81c7 Merge branch 'master' into advanced_logging 2020-07-06 16:52:11 -05:00
Alex Stoken e18e6811dc Merge branch 'master' into advanced_logging 2020-06-27 10:13:03 -05:00
Alex Stoken de191655e4 Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor 2020-06-24 17:21:54 -05:00
Alex Stoken 2d396bea00 Fix bug in --help from percent sign in help string 2020-06-24 16:57:12 -05:00
Alex Stoken d1ca6f231d Delete equip_hyp.yaml 2020-06-24 11:22:12 -05:00
Alex Stoken 611aacf1bf Turn opt into dictionary before sending it to yaml 2020-06-24 10:49:08 -05:00
Alex Stoken bc4ef4861b Default optimizer SGD 2020-06-24 10:07:43 -05:00
Alex Stoken 7abf202cad Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer 2020-06-24 10:03:21 -05:00
Alex Stoken 7edbf6570e Fix help message for cfg files 2020-06-24 09:45:57 -05:00
Alex Stoken d64ad0fbf3 Remove --resume functionality and related checks/logic. 2020-06-24 09:17:27 -05:00
Alex Stoken ccf0af18b1 Revert coco128.yaml to initial commit 2020-06-24 09:13:41 -05:00
Alex Stoken e572bb0803 Add plot_results save location to log_dir 2020-06-21 09:36:28 -05:00
Alex Stoken c8152c81a6 Syntax fixes 2020-06-17 16:32:13 -05:00
Alex Stoken 9b7386f603 Add save_dir arg to test.test, use arg as location for saving batch jpgs 2020-06-17 16:08:46 -05:00
Alex Stoken 945307beba Add save_dir to plot_lr_scheduler and plot_labels
Set save_dir = log_dir in train.py
2020-06-17 16:03:18 -05:00
Alex Stoken 3b2b330872 Move results.txt from weights/ to log_dir 2020-06-17 15:55:45 -05:00
Alex Stoken ade023cff2 Fix hyp file read in and dict update.
Add example of hyp yaml
2020-06-17 10:59:20 -05:00
Alex Stoken 3263a204ea Fix get_latest_run to search 'search_dir' recursivly 2020-06-17 10:34:37 -05:00
Alex Stoken d34291733b Fix get_latest_run() to search inside 'weights' subfolders 2020-06-17 10:26:55 -05:00
Alex Stoken 5f2eeba233 remove old print statements 2020-06-16 17:09:39 -05:00
Alex Stoken 333f678b37 add update default hyp dict with provided yaml 2020-06-16 16:36:20 -05:00
Alex Stoken a448c3bcd7 add logic for resuming and getting hyp for resume run 2020-06-16 16:30:12 -05:00
Alex Stoken 25e51bcec7 add util function to get most recent last.pt file
added logic in train.py __main__ to handle resuming from a run
2020-06-16 15:50:27 -05:00
Alex Stoken 490f1e7b9c add save_dir arg to plot_lr_scheduler, default to current dir.
Uncomment plot_lr_scheduler in train() and pass log_dir as save location
2020-06-16 15:13:03 -05:00
Alex Stoken 4418809cf5 change weights dir (wdir) to be unique to each run, under log_dir 2020-06-16 15:09:51 -05:00
Alex Stoken d9f446cd81 add save yaml of opt and hyp to tensorboard log_dir in train() 2020-06-16 15:06:13 -05:00
Alex Stoken a85e6d0fc0 add parser arg for hyp yaml file 2020-06-16 14:53:32 -05:00