recoilme commited on
Commit
58164da
·
1 Parent(s): e693253
samples/unet_320x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 8e371d1c0f622f780d7ddd91707731d323b45826715b3c6caf820d6d5c79ce0a
  • Pointer size: 131 Bytes
  • Size of remote file: 104 kB

Git LFS Details

  • SHA256: 4ac6bd7cf38aee85221218d46f5b9d459ef21c8078e92119f3d88add2cac86cf
  • Pointer size: 130 Bytes
  • Size of remote file: 95.7 kB
samples/unet_384x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 410ff3f4647c1894b6a5f2a4949a7b72bacf3b718500ca105ee76386f18a9b3e
  • Pointer size: 130 Bytes
  • Size of remote file: 88.5 kB

Git LFS Details

  • SHA256: 246d9e1431e6cafb911b07b9f161610dd9bcd7ab3bea308739ee05a9476edb64
  • Pointer size: 130 Bytes
  • Size of remote file: 88.5 kB
samples/unet_448x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 8cffe81392a800a1ce357d3cce5491142ede85e29477670b8a2b13bf76d52fba
  • Pointer size: 131 Bytes
  • Size of remote file: 116 kB

Git LFS Details

  • SHA256: 448bb0b7018fd4502b3934c34e8278eb112d30c5aadc4bb38ca202a35d4d73a4
  • Pointer size: 131 Bytes
  • Size of remote file: 116 kB
samples/unet_512x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 19875b118e0c5c8c161778f2a2cae4a2a4a098f7a1a3c0939c37e20396e6eeed
  • Pointer size: 131 Bytes
  • Size of remote file: 148 kB

Git LFS Details

  • SHA256: e59134aa87eaa7eefbc95bd7cb7af8e2869568845bfd3da0dcde2964ab60e500
  • Pointer size: 131 Bytes
  • Size of remote file: 153 kB
samples/unet_576x320_0.jpg CHANGED

Git LFS Details

  • SHA256: 3f02ae6da8dacdc8bc65d6ba2c4badeb52b09504c428475b29e56211235e6d2b
  • Pointer size: 130 Bytes
  • Size of remote file: 79.3 kB

Git LFS Details

  • SHA256: ae6bbc972c378cc65f6ff55dd2228539bc6421994e9de0a3b626fe67c1723a4f
  • Pointer size: 130 Bytes
  • Size of remote file: 77.9 kB
samples/unet_576x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 10a8fec9a8c69e8f90b7f95319800f491d3133260505dae5c33994ec16885512
  • Pointer size: 130 Bytes
  • Size of remote file: 37.5 kB

Git LFS Details

  • SHA256: 272c19874dd7a17729aeccbe034df3b52c214a0ff205537ae466453a19bb3125
  • Pointer size: 130 Bytes
  • Size of remote file: 41.1 kB
samples/unet_576x448_0.jpg CHANGED

Git LFS Details

  • SHA256: 3c489543dbb15996f8a424c935fc3016b08135a2e87eb4841cc8489481b79c59
  • Pointer size: 131 Bytes
  • Size of remote file: 109 kB

Git LFS Details

  • SHA256: 45f32251d17e91ddead4db2cf4d462f77b25b88378ad365a0327399abebc5db9
  • Pointer size: 130 Bytes
  • Size of remote file: 98.6 kB
samples/unet_576x512_0.jpg CHANGED

Git LFS Details

  • SHA256: e0201889c4d2ccd0c1dcfa983078514aeaa167610f666c3bf3bc1179471dface
  • Pointer size: 131 Bytes
  • Size of remote file: 111 kB

Git LFS Details

  • SHA256: 6140381a367eb672cd6f661550ec89172a81d9ba9beed478b5da41b93ba45d84
  • Pointer size: 131 Bytes
  • Size of remote file: 104 kB
samples/unet_576x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 5717660ff7c4ffd85ce20632acac9155ca9705081a12d6957de5fa507460e8e7
  • Pointer size: 131 Bytes
  • Size of remote file: 135 kB

Git LFS Details

  • SHA256: 3283a3bede1321acdd9f3e686363508cab482b5c44dd450a21af393a52d187dd
  • Pointer size: 131 Bytes
  • Size of remote file: 144 kB
train.py CHANGED
@@ -931,7 +931,7 @@ for epoch in range(start_epoch, start_epoch + num_epochs):
931
  batch_grads.append(grad)
932
 
933
  # Логируем в Wandb
934
- if use_wandb:
935
  wandb.log({
936
  "mse_loss": loss.detach().item(),
937
  "learning_rate": current_lr,
 
931
  batch_grads.append(grad)
932
 
933
  # Логируем в Wandb
934
+ if use_wandb and accelerator.sync_gradients:
935
  wandb.log({
936
  "mse_loss": loss.detach().item(),
937
  "learning_rate": current_lr,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:331fd7ac0599509b191689032ee7343232b8453954a3225fd06a9fc19db5fc84
3
  size 7014306128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08a4ae9e21fdb572a93caedbe9444e29e1a84986e8d5bcefc5b5d75d86d542ee
3
  size 7014306128