deep copying model

matsengrp · Jun 4, 2024 · 40fbd5f · 40fbd5f
1 parent 7ee3cd5
commit 40fbd5f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/netam/dnsm.py b/netam/dnsm.py
@@ -8,6 +8,7 @@
 
 """
 
+import copy
 import multiprocessing as mp
 
 import torch
@@ -418,7 +419,7 @@ def worker_optimize_branch_length(model, dataset, optimization_kwargs):
     """
     The worker used for parallel branch length optimization.
     """
-    burrito = DNSMBurrito(None, dataset, model)
+    burrito = DNSMBurrito(None, dataset, copy.deepcopy(model))
     return burrito.serial_find_optimal_branch_lengths(dataset, **optimization_kwargs)