yzhouchen001 commited on
Commit
bb73124
·
1 Parent(s): 19a4dfc

model update

Browse files
Files changed (1) hide show
  1. app_utils/model_utils.py +2 -2
app_utils/model_utils.py CHANGED
@@ -15,7 +15,7 @@ lg.setLevel(RDLogger.CRITICAL)
15
  # Load model and data
16
 
17
  def load_model_components():
18
- param_pth = '/data/yzhouc01/FILIP-MS/experiments/20250913_optimized_filip-model/lightning_logs/version_0/hparams.yaml'
19
  with open(param_pth) as f:
20
  params = yaml.load(f, Loader=yaml.FullLoader)
21
 
@@ -24,7 +24,7 @@ def load_model_components():
24
 
25
  # load model
26
 
27
- checkpoint_pth = "/data/yzhouc01/FILIP-MS/experiments/20250913_optimized_filip-model/epoch=1993-train_loss=0.10.ckpt"
28
  params['checkpoint_pth'] = checkpoint_pth
29
  model = get_model(params['model'], params)
30
 
 
15
  # Load model and data
16
 
17
  def load_model_components():
18
+ param_pth = 'hparams.yaml'
19
  with open(param_pth) as f:
20
  params = yaml.load(f, Loader=yaml.FullLoader)
21
 
 
24
 
25
  # load model
26
 
27
+ checkpoint_pth = "pretrained_models/flare.ckpt"
28
  params['checkpoint_pth'] = checkpoint_pth
29
  model = get_model(params['model'], params)
30