diff --git a/dlas b/dlas index 0ef8ab6..fd66c41 160000 --- a/dlas +++ b/dlas @@ -1 +1 @@ -Subproject commit 0ef8ab6872813d1021d4d75e82b63377d28f5a06 +Subproject commit fd66c4104b10a1b794c1b99de034fd5cbafd4078 diff --git a/src/train.py b/src/train.py index 504fef9..bd0656a 100755 --- a/src/train.py +++ b/src/train.py @@ -52,6 +52,7 @@ def train(yaml, launcher='none'): trainer.do_training() if __name__ == "__main__": + # simple check because I'm brain damaged and forgot I can't modify what a module exports by simply changing the booleans that decide what it exports after the fact try: import torch_intermediary if torch_intermediary.OVERRIDE_ADAM: