recoilme commited on
Commit
fd73445
·
1 Parent(s): 16ace30
media/result_grid.jpg CHANGED

Git LFS Details

  • SHA256: 6d9c2c2c031775c0bf23411f34795d17f49990f5f52f48f6048b7bef2bab2d57
  • Pointer size: 132 Bytes
  • Size of remote file: 4.85 MB

Git LFS Details

  • SHA256: 74f962724be2dde2de5c31773d69da1c9c84734b32fdc7013d5f68a0c8510b79
  • Pointer size: 132 Bytes
  • Size of remote file: 4.86 MB
samples/unet_320x640_0.jpg DELETED

Git LFS Details

  • SHA256: c9929b85183b46a9b56fc82cfa9a3b3d09a3cad9144ffeb7a757eb3008faf26f
  • Pointer size: 130 Bytes
  • Size of remote file: 74 kB
samples/unet_384x640_0.jpg DELETED

Git LFS Details

  • SHA256: bbe4d6bb7767a9677a4f597010a94f426b711b5f52644fa8ac9c188bd99f67c6
  • Pointer size: 131 Bytes
  • Size of remote file: 154 kB
samples/unet_448x640_0.jpg DELETED

Git LFS Details

  • SHA256: 8740815bd5d32d21e8cef4ac2fb9680ab1f52db09e6546d1ffa78254303478f9
  • Pointer size: 130 Bytes
  • Size of remote file: 89.3 kB
samples/unet_512x640_0.jpg DELETED

Git LFS Details

  • SHA256: 700aa94e24c559da490a1158f0f1917534f427e3d7f3474079fcbfc61e3214a9
  • Pointer size: 131 Bytes
  • Size of remote file: 155 kB
samples/unet_576x640_0.jpg DELETED

Git LFS Details

  • SHA256: d5476a03010634c4fb8231a8e3b4af146cd8a56d04af72a811ce7d029e0ec392
  • Pointer size: 130 Bytes
  • Size of remote file: 36.3 kB
samples/unet_640x320_0.jpg DELETED

Git LFS Details

  • SHA256: 107fde42a18cc3d168e1fe0f0f399c12d647cc1e7dac188c0af0edfd2b052696
  • Pointer size: 130 Bytes
  • Size of remote file: 14.7 kB
samples/unet_640x384_0.jpg DELETED

Git LFS Details

  • SHA256: e9d60664fc9533b2043cebcd23c05cea36e91b2abd58ffc0886f8388207d67dd
  • Pointer size: 130 Bytes
  • Size of remote file: 88.1 kB
samples/unet_640x448_0.jpg DELETED

Git LFS Details

  • SHA256: b9bb378e30180c8a6d43bf0affbf4182ad9ca99bdff7d3fdc2859d77d1fc2cb8
  • Pointer size: 130 Bytes
  • Size of remote file: 25 kB
samples/unet_640x512_0.jpg DELETED

Git LFS Details

  • SHA256: ac922f30b0f8294308374a2d4eb981d565b2f9339ecbb38c2ba7b4a29a672c0c
  • Pointer size: 131 Bytes
  • Size of remote file: 135 kB
samples/unet_640x576_0.jpg DELETED

Git LFS Details

  • SHA256: 2ed4552767ee98f0cf7ff74040bb17be2f48239cbbf7a4590d6128574fb3c045
  • Pointer size: 131 Bytes
  • Size of remote file: 119 kB
samples/unet_640x640_0.jpg DELETED

Git LFS Details

  • SHA256: cd8641e8153b9d2eb064e459593c2dcf4c3b404cfafb8aa93ea5c90eb82286be
  • Pointer size: 131 Bytes
  • Size of remote file: 148 kB
test.ipynb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c2b00b404e66b2ee215280298d3e670b5c6d7ff7d70075052011f4f7719973f5
3
- size 4598308
 
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/640"
30
  project = "unet"
31
- batch_size = 256
32
  base_learning_rate = 3e-5
33
  min_learning_rate = 2.5e-5
34
  num_epochs = 10
35
- sample_interval_share = 40
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:402b5747329ecd5573887b46e35e687b8f3d6c79ccba522c06e58082d0eace87
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:9437f6d40639f1ad6d95d6586009d7b1bf0c8e99959d29da5c9c9645cae39ea3
3
- size 1899
 
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:22402fe3c630ab551173c2bd11af4a7ae4ed76c04f001535f3791cd82da93719
3
- size 3103078992
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe4cc15557dac5284d2e197c38a1ae2eb0b8154bf994e9f5bac7e85710067f74
3
+ size 3302474272