Update README.md
Browse files
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>/
|
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>/
|
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>/
|
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>/
|
65 |
|
66 |
# For segmentation
|
67 |
-
python mmsegmentation/tools/my_test.py mmsegmentation/config/<dataset_name>/
|
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
|