recoilme commited on
Commit
f52c2f8
·
1 Parent(s): 898bc03
samples/unet_320x640_0.jpg ADDED

Git LFS Details

  • SHA256: de70cc7677216a340eb62bf3eada0d6cad482256b5732cb4b2f88254df756773
  • Pointer size: 130 Bytes
  • Size of remote file: 67.7 kB
samples/unet_384x640_0.jpg ADDED

Git LFS Details

  • SHA256: 8a7e4b4ccf58eb40cd74596425d594f0a76bc0c1d33476fb59e28a5e137e3459
  • Pointer size: 131 Bytes
  • Size of remote file: 164 kB
samples/unet_448x640_0.jpg ADDED

Git LFS Details

  • SHA256: 77d5da2186f63a0417dbdeb892d9c18a0012922d56bab0776e10d9b7ecf519dc
  • Pointer size: 131 Bytes
  • Size of remote file: 112 kB
samples/unet_512x640_0.jpg ADDED

Git LFS Details

  • SHA256: 3a7d8e22794795e8b40076dc403729ec64ed04aed57d6614aee57defb9dd0088
  • Pointer size: 131 Bytes
  • Size of remote file: 160 kB
samples/unet_576x640_0.jpg ADDED

Git LFS Details

  • SHA256: 49de4a3333f896ffe4f8766fb80a20f7bf3bccd4cb3bed3d7ff499801aab2bbd
  • Pointer size: 131 Bytes
  • Size of remote file: 118 kB
samples/unet_640x320_0.jpg ADDED

Git LFS Details

  • SHA256: 501741e5bd0aa1cb01c0b826b10544b2413eb019527783534179f74dbb4bb99d
  • Pointer size: 130 Bytes
  • Size of remote file: 58.8 kB
samples/unet_640x384_0.jpg ADDED

Git LFS Details

  • SHA256: ad552e368f7ba04bc4274b86464fae1d80efe29c3e91b9ea5ff2a18b4463023f
  • Pointer size: 130 Bytes
  • Size of remote file: 59.7 kB
samples/unet_640x448_0.jpg ADDED

Git LFS Details

  • SHA256: 4c8cf68d5afb65ae0d0f71a7d7a94be39c8cd7973e0c02b863493f1d48e24ed2
  • Pointer size: 130 Bytes
  • Size of remote file: 97.7 kB
samples/unet_640x512_0.jpg ADDED

Git LFS Details

  • SHA256: ec08ac216bba512d71c75b115506d189f7597782368c0c6371c5d6725bcce44a
  • Pointer size: 130 Bytes
  • Size of remote file: 86.6 kB
samples/unet_640x576_0.jpg ADDED

Git LFS Details

  • SHA256: 71c327a4d8a4ac4e071d03a1855150dc0103d332579a3fb1b758f69a139678f3
  • Pointer size: 131 Bytes
  • Size of remote file: 172 kB
samples/unet_640x640_0.jpg ADDED

Git LFS Details

  • SHA256: 69d1d68c72172b179048c6c7272a8b0c778c3b263da4ff17f3d347daca4079ad
  • Pointer size: 130 Bytes
  • Size of remote file: 78.6 kB
train.py CHANGED
@@ -26,12 +26,12 @@ import torch.nn.functional as F
26
  from collections import deque
27
 
28
  # --------------------------- Параметры ---------------------------
29
- ds_path = "/workspace/sdxs3d/datasets/640"
30
  project = "unet"
31
  batch_size = 48
32
  base_learning_rate = 5e-5
33
  min_learning_rate = 1e-5
34
- num_epochs = 40
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:636ed60d2a73b31fe88849234ff578446cd018092fdc9210575893551220bfc3
3
- size 1887
 
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:b8ec1c74913276f691c838b5a85cd65939c4437c0ce1741f69c8c8931dd39112
3
- size 3092571208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1443f2abd54e2b73c333d8556c0dc008a2f08f7a5e2823a553347f1eebc3cd61
3
+ size 6184944280