bugfix on takeweightsfrom option

DL4Jets · Jan 23, 2022 · a5df438 · a5df438
1 parent a112aff
commit a5df438
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/training/training_base.py b/training/training_base.py
@@ -29,7 +29,7 @@ def __init__(
 				useweights=False, testrun=False,
                 testrun_fraction=0.1,
 				resumeSilently=False, 
-				renewtokens=True,
+				renewtokens=False,
 				collection_class=DataCollection,
 				parser=None,
                 recreate_silently=False
@@ -241,11 +241,9 @@ def setModel(self,model,**modelargs):
             self.keras_model.build(None)
 
         if len(self.keras_weight_model_path):
-            from DeepJetCore.modeltools import apply_weights_where_possible
-            tbcopy = self
-            tbcopy.loadModel(self.keras_weight_model_path)
+            from DeepJetCore.modeltools import apply_weights_where_possible, load_model
             self.keras_model = apply_weights_where_possible(self.keras_model, 
-                                         tbcopy.keras_model)
+                                         load_model(self.keras_weight_model_path))
         #try:
         #    self.keras_model=model(self.keras_inputs,**modelargs)
         #except BaseException as e:
@@ -358,7 +356,7 @@ def _initTraining(self,
         from _thread import start_new_thread
 
         if self.renewtokens:
-            print('starting afs backgrounder')
+            print('afs backgrounder has proven to be unreliable, use with care')
             checkTokens()
             start_new_thread(renew_token_process,())