768
Browse files- media/result_grid.jpg +2 -2
- samples/unet_320x640_0.jpg +0 -3
- samples/unet_384x640_0.jpg +0 -3
- samples/unet_448x640_0.jpg +0 -3
- samples/unet_512x640_0.jpg +0 -3
- samples/unet_576x640_0.jpg +0 -3
- samples/unet_640x320_0.jpg +0 -3
- samples/unet_640x384_0.jpg +0 -3
- samples/unet_640x448_0.jpg +0 -3
- samples/unet_640x512_0.jpg +0 -3
- samples/unet_640x576_0.jpg +0 -3
- samples/unet_640x640_0.jpg +0 -3
- test.ipynb +2 -2
- train.py +3 -3
- unet/diffusion_pytorch_model.safetensors +1 -1
- {unet_very_old → unet640}/config.json +2 -2
- {unet_very_old → unet640}/diffusion_pytorch_model.safetensors +2 -2
media/result_grid.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_320x640_0.jpg
DELETED
Git LFS Details
|
samples/unet_384x640_0.jpg
DELETED
Git LFS Details
|
samples/unet_448x640_0.jpg
DELETED
Git LFS Details
|
samples/unet_512x640_0.jpg
DELETED
Git LFS Details
|
samples/unet_576x640_0.jpg
DELETED
Git LFS Details
|
samples/unet_640x320_0.jpg
DELETED
Git LFS Details
|
samples/unet_640x384_0.jpg
DELETED
Git LFS Details
|
samples/unet_640x448_0.jpg
DELETED
Git LFS Details
|
samples/unet_640x512_0.jpg
DELETED
Git LFS Details
|
samples/unet_640x576_0.jpg
DELETED
Git LFS Details
|
samples/unet_640x640_0.jpg
DELETED
Git LFS Details
|
test.ipynb
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:3e89d6273e5bed930830e22ad4c43a8ac63a0737fa1eefacb085e8b13dd014f3
|
| 3 |
+
size 7281307
|
train.py
CHANGED
|
@@ -26,13 +26,13 @@ from collections import deque
|
|
| 26 |
from transformers import AutoTokenizer, AutoModel
|
| 27 |
|
| 28 |
# --------------------------- Параметры ---------------------------
|
| 29 |
-
ds_path = "/workspace/sdxs/datasets/
|
| 30 |
project = "unet"
|
| 31 |
-
batch_size =
|
| 32 |
base_learning_rate = 3e-5
|
| 33 |
min_learning_rate = 2.5e-5
|
| 34 |
num_epochs = 10
|
| 35 |
-
sample_interval_share =
|
| 36 |
max_length = 192
|
| 37 |
use_wandb = True
|
| 38 |
use_comet_ml = False
|
|
|
|
| 26 |
from transformers import AutoTokenizer, AutoModel
|
| 27 |
|
| 28 |
# --------------------------- Параметры ---------------------------
|
| 29 |
+
ds_path = "/workspace/sdxs/datasets/768"
|
| 30 |
project = "unet"
|
| 31 |
+
batch_size = 384
|
| 32 |
base_learning_rate = 3e-5
|
| 33 |
min_learning_rate = 2.5e-5
|
| 34 |
num_epochs = 10
|
| 35 |
+
sample_interval_share = 20
|
| 36 |
max_length = 192
|
| 37 |
use_wandb = True
|
| 38 |
use_comet_ml = False
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 6604736640
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:358904c62d6323f4840f74a3d454d936dcefda7b190c07352dbc2f9d4ad31be9
|
| 3 |
size 6604736640
|
{unet_very_old → unet640}/config.json
RENAMED
|
@@ -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:6d46560eb07a34ba817630209fa68df908c453a1adc295b08ef9f6aaec074de9
|
| 3 |
+
size 1895
|
{unet_very_old → unet640}/diffusion_pytorch_model.safetensors
RENAMED
|
@@ -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:fe4cc15557dac5284d2e197c38a1ae2eb0b8154bf994e9f5bac7e85710067f74
|
| 3 |
+
size 3302474272
|