youngzhou12 commited on
Commit
e29b149
·
verified ·
1 Parent(s): d891cb8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -40,20 +40,20 @@ Please follow the [instruction](https://github.com/yangzhou12/BenchX/blob/releas
40
  To fine-tune MRM for classification, run this command:
41
 
42
  ```
43
- python bin/train.py config/classification/<dataset_name>/MRM.yml
44
  ```
45
 
46
  ### 2. Segmentation
47
  To fine-tune MRM for segmentation, run this command:
48
 
49
  ```
50
- python mmsegmentation/tools/train.py config/benchmark/<dataset_name>/MRM.yml
51
  ```
52
 
53
  ### 3. Report Generation
54
  To fine-tune MRM for report generation, run this command:
55
  ```
56
- python bin/train.py config/report_generation/<dataset_name>/MRM.yml
57
  ```
58
 
59
  ### 4. Evaluation
@@ -61,10 +61,10 @@ To evaluate fine-tuned MRM models, run:
61
 
62
  ```
63
  # For classification and report generation
64
- python bin/test.py config/<task_name>/<dataset_name>/MRM.yml validator.splits=[test] ckpt_dir=<path_to_checkpoint>
65
 
66
  # For segmentation
67
- python mmsegmentation/tools/my_test.py mmsegmentation/config/<dataset_name>/MRM.yml <path_to_checkpoint>
68
  ```
69
 
70
  ## Citations
 
40
  To fine-tune MRM for classification, run this command:
41
 
42
  ```
43
+ python bin/train.py config/classification/<dataset_name>/mrm.yml
44
  ```
45
 
46
  ### 2. Segmentation
47
  To fine-tune MRM for segmentation, run this command:
48
 
49
  ```
50
+ python mmsegmentation/tools/train.py config/benchmark/<dataset_name>/mrm.yml
51
  ```
52
 
53
  ### 3. Report Generation
54
  To fine-tune MRM for report generation, run this command:
55
  ```
56
+ python bin/train.py config/report_generation/<dataset_name>/mrm.yml
57
  ```
58
 
59
  ### 4. Evaluation
 
61
 
62
  ```
63
  # For classification and report generation
64
+ python bin/test.py config/<task_name>/<dataset_name>/mrm.yml validator.splits=[test] ckpt_dir=<path_to_checkpoint>
65
 
66
  # For segmentation
67
+ python mmsegmentation/tools/my_test.py mmsegmentation/config/<dataset_name>/mrm.yml <path_to_checkpoint>
68
  ```
69
 
70
  ## Citations