2511
Browse files
train.py
CHANGED
|
@@ -29,8 +29,8 @@ from collections import deque
|
|
| 29 |
ds_path = "/workspace/sdxs/datasets/640"
|
| 30 |
project = "unet"
|
| 31 |
batch_size = 48
|
| 32 |
-
base_learning_rate =
|
| 33 |
-
min_learning_rate =
|
| 34 |
num_epochs = 50
|
| 35 |
# samples/save per epoch
|
| 36 |
sample_interval_share = 5
|
|
|
|
| 29 |
ds_path = "/workspace/sdxs/datasets/640"
|
| 30 |
project = "unet"
|
| 31 |
batch_size = 48
|
| 32 |
+
base_learning_rate = 4e-5
|
| 33 |
+
min_learning_rate = 2e-5
|
| 34 |
num_epochs = 50
|
| 35 |
# samples/save per epoch
|
| 36 |
sample_interval_share = 5
|