Commit a9cf5c38 authored by eric_hammy's avatar eric_hammy
Browse files

Update polybeast_learner.py

parent 51ba73c1
......@@ -515,3 +515,6 @@ def main(flags):
train(flags)
elif flags.mode.startswith("test"):
test(flags)
if flags.wandb:
wandb.finish()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment