Skip to content
Permalink
Browse files

Update model_PACS.py

  • Loading branch information...
liyiying committed May 13, 2019
1 parent 26cef81 commit e9e52b661dad5828b13edd0842833aba50c04c05
Showing with 3 additions and 3 deletions.
  1. +3 −3 model_PACS.py
@@ -301,7 +301,7 @@ def init_dg_function(self,flags):
def train(self, flags):

write_log(flags, self.flags_log)
self.pre_train()
self.pre_train(flags)
self.reinit_network_P(flags)
time_start = datetime.datetime.now()

@@ -438,8 +438,8 @@ def train(self, flags):
self.validate_workflow(self.batImageGenVals, flags, _)
torch.cuda.empty_cache()

def pre_train(self):
model_path = os.path.join('model_output/PACS/baseline/', 'best_model.tar')
def pre_train(self, flags):
model_path = os.path.join(flags.load_path, 'best_model.tar')
if os.path.exists(model_path):
self.load_state_dict(state_dict=model_path)

0 comments on commit e9e52b6

Please sign in to comment.
You can’t perform that action at this time.