Update README.md
Browse files
README.md
CHANGED
@@ -6,8 +6,9 @@ base_model:
|
|
6 |
- facebook/sam-vit-base
|
7 |
---
|
8 |
|
9 |
-
|
10 |
|
|
|
11 |
DATASET:
|
12 |
TRAIN_PATH: "./ufpr_dataset/training"
|
13 |
TEST_PATH: "./ufpr_dataset/testing"
|
@@ -21,4 +22,4 @@ TRAIN:
|
|
21 |
BATCH_SIZE: 1
|
22 |
NUM_EPOCHS: 30
|
23 |
LEARNING_RATE: 0.0001
|
24 |
-
PATIENCE: 10
|
|
|
6 |
- facebook/sam-vit-base
|
7 |
---
|
8 |
|
9 |
+
### Parameters on which the model was trained
|
10 |
|
11 |
+
```yaml
|
12 |
DATASET:
|
13 |
TRAIN_PATH: "./ufpr_dataset/training"
|
14 |
TEST_PATH: "./ufpr_dataset/testing"
|
|
|
22 |
BATCH_SIZE: 1
|
23 |
NUM_EPOCHS: 30
|
24 |
LEARNING_RATE: 0.0001
|
25 |
+
PATIENCE: 10
|