1007
Browse files- samples/unet_320x576_0.jpg +2 -2
- samples/unet_384x576_0.jpg +2 -2
- samples/unet_448x576_0.jpg +2 -2
- samples/unet_512x576_0.jpg +2 -2
- samples/unet_576x320_0.jpg +2 -2
- samples/unet_576x384_0.jpg +2 -2
- samples/unet_576x448_0.jpg +2 -2
- samples/unet_576x512_0.jpg +2 -2
- samples/unet_576x576_0.jpg +2 -2
- train.py +1 -1
- unet/diffusion_pytorch_model.safetensors +1 -1
samples/unet_320x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_384x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_448x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_512x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x384_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x448_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x512_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/unet_576x576_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
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:
|
| 3 |
size 7014306128
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:08a4ae9e21fdb572a93caedbe9444e29e1a84986e8d5bcefc5b5d75d86d542ee
|
| 3 |
size 7014306128
|