Hugging Face's logo

AiArtLab
/
sdxs3d

Diffusers
Safetensors
Model card Files
xet
 Community
recoilme commited on
Commit
d30a126
·
1 Parent(s): 82b8cf4

640

Browse files
Files changed (20) hide show
  1. .gitignore +1 -1
  2. samples/unet_192x384_0.jpg +2 -2
  3. samples/unet_256x384_0.jpg +2 -2
  4. samples/unet_320x384_0.jpg +2 -2
  5. samples/unet_384x192_0.jpg +2 -2
  6. samples/unet_384x256_0.jpg +2 -2
  7. samples/unet_384x320_0.jpg +2 -2
  8. samples/unet_384x384_0.jpg +2 -2
  9. samples/unet_384x640_0.jpg +3 -0
  10. samples/unet_448x640_0.jpg +3 -0
  11. samples/unet_512x640_0.jpg +3 -0
  12. samples/unet_576x640_0.jpg +3 -0
  13. samples/unet_640x384_0.jpg +3 -0
  14. samples/unet_640x448_0.jpg +3 -0
  15. samples/unet_640x512_0.jpg +3 -0
  16. samples/unet_640x576_0.jpg +3 -0
  17. samples/unet_640x640_0.jpg +3 -0
  18. src/dataset_sample.ipynb +0 -0
  19. train.py +4 -4
  20. unet/diffusion_pytorch_model.safetensors +1 -1
.gitignore CHANGED
@@ -8,7 +8,7 @@ src/samples
8
  # cache
9
  cache
10
  datasets/mjnj
11
- # datasets
12
  test
13
  wandb
14
  nohup.out
 
8
  # cache
9
  cache
10
  datasets/mjnj
11
+ datasets/640
12
  test
13
  wandb
14
  nohup.out
samples/unet_192x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 2cfd2fe5f925ea0e855b516dc91eb8c2fd47f3191ae191c9a24a74d0b7b7edf7
  • Pointer size: 130 Bytes
  • Size of remote file: 43 kB

Git LFS Details

  • SHA256: 5fffec8dd4945df30c4eb4e84f2ae5f58da70451a2d3abedfabc7fde1fac1481
  • Pointer size: 130 Bytes
  • Size of remote file: 44 kB
samples/unet_256x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 4aaaeeb36479c94837e93d0b82bf965fcf7df84a9b42191d54e38fd2dbec4c50
  • Pointer size: 130 Bytes
  • Size of remote file: 46.2 kB

Git LFS Details

  • SHA256: 71a25ba3709a024303c7678ba203fb33cac38a6c81c846f12a9666c11468068e
  • Pointer size: 130 Bytes
  • Size of remote file: 53.2 kB
samples/unet_320x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 3765133fa41cf0c363000959ae906dc7fe9698a6b3ab4cca4572763a13ca8853
  • Pointer size: 130 Bytes
  • Size of remote file: 59.7 kB

Git LFS Details

  • SHA256: d1d0420bff4bd20deac5a093658b5f63c1e6b8e07ea7fee16e45e792e65a2112
  • Pointer size: 130 Bytes
  • Size of remote file: 58.4 kB
samples/unet_384x192_0.jpg CHANGED

Git LFS Details

  • SHA256: 98800aa01af951838fa34e5a7530373667b25112dd002e0a1e1d11531ac9e112
  • Pointer size: 130 Bytes
  • Size of remote file: 48.8 kB

Git LFS Details

  • SHA256: 9099773283d326e3cb25d71b87d20681df807e19245600807f52485e0e801045
  • Pointer size: 130 Bytes
  • Size of remote file: 51.2 kB
samples/unet_384x256_0.jpg CHANGED

Git LFS Details

  • SHA256: 7499ef6eb758f8d1983522eb5173ca0521c92772bdcd23d87632928007b1dc8e
  • Pointer size: 130 Bytes
  • Size of remote file: 55.5 kB

Git LFS Details

  • SHA256: 5f656f6b7fa904023219c9f83e42e0ce2a57485f3d2b0fe04e58d9a05f4ffcf8
  • Pointer size: 130 Bytes
  • Size of remote file: 68 kB
samples/unet_384x320_0.jpg CHANGED

Git LFS Details

  • SHA256: e3f9019e2e79ff17abc25c20ba3c78c9cdc7e1a2c38056560a9fb164126eebca
  • Pointer size: 130 Bytes
  • Size of remote file: 85 kB

Git LFS Details

  • SHA256: 87ef63669ff8918a1af1b786e5dc4ab6b3b21a10510905a38536a992e05681f1
  • Pointer size: 130 Bytes
  • Size of remote file: 85 kB
samples/unet_384x384_0.jpg CHANGED

Git LFS Details

  • SHA256: d46731081d2295c40c68236c120a45c36324092808c4ef3cf7a02c138b9a8706
  • Pointer size: 130 Bytes
  • Size of remote file: 71.3 kB

Git LFS Details

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

Git LFS Details

  • SHA256: 097ac7366b92bd92451bed4bdfd238f3e98d1369fea17c9d4babae5bc25af5e6
  • Pointer size: 131 Bytes
  • Size of remote file: 128 kB
samples/unet_448x640_0.jpg ADDED

Git LFS Details

  • SHA256: 06bf9945c0c12cf170ef7e0291eb8d9c42722418a729d7dbbd2830b8bc591f5d
  • Pointer size: 130 Bytes
  • Size of remote file: 78.1 kB
samples/unet_512x640_0.jpg ADDED

Git LFS Details

  • SHA256: 09f05a3f07aa1a1be4cdb6f8b7a3ae0dfa94f2e1c205026fa4c0f9c19f8f7293
  • Pointer size: 131 Bytes
  • Size of remote file: 192 kB
samples/unet_576x640_0.jpg ADDED

Git LFS Details

  • SHA256: cfc6b0cfbb01076a012cc8043bb2ea16792f65588a0d42df6a4e38be7d3b02a7
  • Pointer size: 131 Bytes
  • Size of remote file: 178 kB
samples/unet_640x384_0.jpg ADDED

Git LFS Details

  • SHA256: 218d44f1dabae7a4db43708af79e585479b2597653d3de1f7c78cc69376f47b8
  • Pointer size: 130 Bytes
  • Size of remote file: 91.4 kB
samples/unet_640x448_0.jpg ADDED

Git LFS Details

  • SHA256: baeb98eddb35043c05f684005e4e5ba754f2cd6baf4c4c083a9ca86f69f9c7f3
  • Pointer size: 131 Bytes
  • Size of remote file: 158 kB
samples/unet_640x512_0.jpg ADDED

Git LFS Details

  • SHA256: 4922b1d091fd35f743a77145d5a0a7ff12853601054cac18ea544ce4e8a84401
  • Pointer size: 130 Bytes
  • Size of remote file: 68.2 kB
samples/unet_640x576_0.jpg ADDED

Git LFS Details

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

Git LFS Details

  • SHA256: daa3583b475b08130805429057d64bfa928c5d0036b8df43d9cd600bd099e765
  • Pointer size: 131 Bytes
  • Size of remote file: 107 kB
src/dataset_sample.ipynb CHANGED
The diff for this file is too large to render. See raw diff
 
train.py CHANGED
@@ -26,11 +26,11 @@ import torch.nn.functional as F
26
  from collections import deque
27
 
28
  # --------------------------- Параметры ---------------------------
29
- ds_path = "/workspace/sdxs3d/datasets/mjnj"
30
  project = "unet"
31
- batch_size = 32
32
- base_learning_rate = 2e-5
33
- min_learning_rate = 6e-6
34
  num_epochs = 100
35
  # samples/save per epoch
36
  sample_interval_share = 2
 
26
  from collections import deque
27
 
28
  # --------------------------- Параметры ---------------------------
29
+ ds_path = "/workspace/sdxs3d/datasets/640"
30
  project = "unet"
31
+ batch_size = 64
32
+ base_learning_rate = 6e-5
33
+ min_learning_rate = 1e-5
34
  num_epochs = 100
35
  # samples/save per epoch
36
  sample_interval_share = 2
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1d4ec32042df3050f3288c966be4051f5bfc9e1c1804a812797815770b1f299c
3
  size 6184944280
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:904f55427886ce9f62f796950c6b6425b42ad02a6f5940f2abc6bb50b33924c9
3
  size 6184944280