Added Model rescale and prepared a release upgrade

2023-08-01 13:49:54 -06:00
parent 63cacf4362
commit 8b8d53888d
15 changed files with 388 additions and 64 deletions
--- a/toolkit/job.py
+++ b/toolkit/job.py
@@ -13,6 +13,9 @@ def get_job(config_path, name=None):
    if job == 'train':
        from jobs import TrainJob
        return TrainJob(config)
+    if job == 'mod':
+        from jobs import ModJob
+        return ModJob(config)

    # elif job == 'train':
    #     from jobs import TrainJob