Made a very basic vae trainer.

2023-07-17 19:03:50 -06:00
parent 78b59c5e99
commit 439310e4dc
11 changed files with 410 additions and 89 deletions
--- a/toolkit/job.py
+++ b/toolkit/job.py
@@ -10,6 +10,10 @@ def get_job(config_path):
    if job == 'extract':
        from jobs import ExtractJob
        return ExtractJob(config)
+    if job == 'train':
+        from jobs import TrainJob
+        return TrainJob(config)
+
    # elif job == 'train':
    #     from jobs import TrainJob
    #     return TrainJob(config)