Merge pull request #344 from anyangml2nd/fix/dp-finetune

Fix: use finetune tag

Author: anyangml

lambench/models/dp_models.py

@@ -75,7 +75,7 @@ def _finetune(model: Path, task: PropertyFinetuneTask):
         os.environ["NUM_WORKERS"] = "0"
         ngpus = torch.cuda.device_count()
         os.system(
-            f"torchrun --no_python --nproc_per_node={ngpus} dp --pt train input.json --skip-neighbor-stat"
+            f"torchrun --no_python --nproc_per_node={ngpus} dp --pt train input.json --finetune {model} --skip-neighbor-stat"
         )
         return Path("model.ckpt.pt")  # hard coded in deepmd-kit
 

lambench/models/models_config.yml

@@ -26,8 +26,8 @@
     num_parameters: 7021990
     packages:
       deepmd-kit: P0131
-  show_finetune_task: True
-  show_calculator_task: True
+  show_direct_task: False # Hide deprecated model
+  skip_tasks: [CalculatorTask, DirectPredictTask, PropertyFinetuneTask]
 
 - model_name: dpa3_250415_v3_1_0_3M
   model_type: DP