Honzus24 commited on
Commit
b115399
·
verified ·
1 Parent(s): 2dc8258

Update configs/train_config.yaml

Browse files
Files changed (1) hide show
  1. configs/train_config.yaml +1 -1
configs/train_config.yaml CHANGED
@@ -39,4 +39,4 @@ inference_args:
39
  3d_model_path: ./models/weights/flexpert_3d_weights.bin
40
  enm_type: 'ANM'
41
  batch_size: 4
42
- prediction_output_dir: 'prediction_results/{}-{}-{}-predictions.txt'
 
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'