youngzhou12 commited on
Commit
41acb92
·
verified ·
1 Parent(s): b5b5b16

Update README.md

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