10
Browse files- samples/unet_320x640_0.jpg +3 -0
- samples/unet_384x640_0.jpg +3 -0
- samples/unet_448x640_0.jpg +3 -0
- samples/unet_512x640_0.jpg +3 -0
- samples/unet_576x640_0.jpg +3 -0
- samples/unet_640x320_0.jpg +3 -0
- samples/unet_640x384_0.jpg +3 -0
- samples/unet_640x448_0.jpg +3 -0
- samples/unet_640x512_0.jpg +3 -0
- samples/unet_640x576_0.jpg +3 -0
- samples/unet_640x640_0.jpg +3 -0
- train.py +2 -2
- unet/config.json +2 -2
- unet/diffusion_pytorch_model.safetensors +2 -2
samples/unet_320x640_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_384x640_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_448x640_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_512x640_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_576x640_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_640x320_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_640x384_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_640x448_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_640x512_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_640x576_0.jpg
ADDED
|
Git LFS Details
|
samples/unet_640x640_0.jpg
ADDED
|
Git LFS Details
|
train.py
CHANGED
|
@@ -26,12 +26,12 @@ import torch.nn.functional as F
|
|
| 26 |
from collections import deque
|
| 27 |
|
| 28 |
# --------------------------- Параметры ---------------------------
|
| 29 |
-
ds_path = "/workspace/
|
| 30 |
project = "unet"
|
| 31 |
batch_size = 48
|
| 32 |
base_learning_rate = 5e-5
|
| 33 |
min_learning_rate = 1e-5
|
| 34 |
-
num_epochs =
|
| 35 |
# samples/save per epoch
|
| 36 |
sample_interval_share = 3
|
| 37 |
use_wandb = True
|
|
|
|
| 26 |
from collections import deque
|
| 27 |
|
| 28 |
# --------------------------- Параметры ---------------------------
|
| 29 |
+
ds_path = "/workspace/sdxs/datasets/640"
|
| 30 |
project = "unet"
|
| 31 |
batch_size = 48
|
| 32 |
base_learning_rate = 5e-5
|
| 33 |
min_learning_rate = 1e-5
|
| 34 |
+
num_epochs = 50
|
| 35 |
# samples/save per epoch
|
| 36 |
sample_interval_share = 3
|
| 37 |
use_wandb = True
|
unet/config.json
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fac879374be0c1cd3ffcbfa188e72f50576f25326aada5da077b0fc7af442ad4
|
| 3 |
+
size 1874
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1443f2abd54e2b73c333d8556c0dc008a2f08f7a5e2823a553347f1eebc3cd61
|
| 3 |
+
size 6184944280
|