youngzhou12 commited on
Commit
ba166a6
·
verified ·
1 Parent(s): d69ba54

Update README.md

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