This commit is contained in:
mrq 2023-02-23 07:20:40 +00:00
parent 941a27d2b3
commit 41fca1a101
2 changed files with 2 additions and 1 deletions

2
dlas

@ -1 +1 @@
Subproject commit 0ef8ab6872813d1021d4d75e82b63377d28f5a06 Subproject commit fd66c4104b10a1b794c1b99de034fd5cbafd4078

View File

@ -52,6 +52,7 @@ def train(yaml, launcher='none'):
trainer.do_training() trainer.do_training()
if __name__ == "__main__": 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: try:
import torch_intermediary import torch_intermediary
if torch_intermediary.OVERRIDE_ADAM: if torch_intermediary.OVERRIDE_ADAM: