Commit Graph
33 Commits
Author SHA1 Message Date
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