Hugging Face's logo

Spaces:
Honzus24
/
flexpert
Running on Zero

App Files Community
1
Honzus24 commited on
Commit
6274465
·
verified ·
1 Parent(s): 26dd42d

Update configs/train_config.yaml

Browse files
Files changed (1) hide show
  1. configs/train_config.yaml +2 -2
configs/train_config.yaml CHANGED
@@ -35,8 +35,8 @@ training_args:
35
 
36
  inference_args:
37
  device: 'cuda:0'
38
- seq_model_path: ./models/weights/flexpert_seq_weights.bin
39
- 3d_model_path: ./models/weights/flexpert_3d_weights.bin
40
  enm_type: 'ANM'
41
  batch_size: 4
42
  prediction_output_dir: 'prediction_results/{}-{}.txt'
 
35
 
36
  inference_args:
37
  device: 'cuda:0'
38
+ seq_model_path: 'flexpert_seq_weights.bin'
39
+ 3d_model_path: 'flexpert_3d_weights.bin'
40
  enm_type: 'ANM'
41
  batch_size: 4
42
  prediction_output_dir: 'prediction_results/{}-{}.txt'