free memory on capture output

HodgeLab · Apr 4, 2022 · 4f0e37a · 4f0e37a
1 parent ab2692b
commit 4f0e37a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/train/train.jl b/src/train/train.jl
@@ -639,6 +639,8 @@ function _capture_output(output_dict, output_directory, id)
             open(joinpath(output_path, key), "w") do io
                 Arrow.write(io, df)
             end
+            df = nothing 
+            GC.gc()
         end
     end
     open(joinpath(output_path, "high_level_outputs"), "w") do io