Commit b84e1382 authored by Alexander Nicholas Leye's avatar Alexander Nicholas Leye
Browse files

clear CUDA cache on training

parent 77b04ff4
......@@ -184,4 +184,5 @@ def main():
if __name__ == '__main__':
torch.cuda.empty_cache()
main()
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