ashhadahsan commited on
Commit
65013fd
·
verified ·
1 Parent(s): 31a1c50

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -6,8 +6,9 @@ base_model:
6
  - facebook/sam-vit-base
7
  ---
8
 
9
- # Parameters on which the model was trained
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