recoilme
commited on
Commit
·
b6d8e34
1
Parent(s):
29ec028
v002
Browse files- result_grid.jpg +2 -2
- src/sample.ipynb +0 -0
- train.py +3 -3
result_grid.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
src/sample.ipynb
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
train.py
CHANGED
|
@@ -28,9 +28,9 @@ from collections import deque
|
|
| 28 |
# --------------------------- Параметры ---------------------------
|
| 29 |
ds_path = "/workspace/sdxs3d/datasets/mjnj"
|
| 30 |
project = "unet"
|
| 31 |
-
batch_size =
|
| 32 |
-
base_learning_rate =
|
| 33 |
-
min_learning_rate =
|
| 34 |
num_epochs = 50
|
| 35 |
# samples/save per epoch
|
| 36 |
sample_interval_share = 5
|
|
|
|
| 28 |
# --------------------------- Параметры ---------------------------
|
| 29 |
ds_path = "/workspace/sdxs3d/datasets/mjnj"
|
| 30 |
project = "unet"
|
| 31 |
+
batch_size = 64
|
| 32 |
+
base_learning_rate = 4e-5
|
| 33 |
+
min_learning_rate = 8e-6
|
| 34 |
num_epochs = 50
|
| 35 |
# samples/save per epoch
|
| 36 |
sample_interval_share = 5
|