Commit ·
2473850
1
Parent(s): f5927bd
Remove logging statements
Browse files
dataset/__pycache__/celeba.cpython-310.pyc
CHANGED
|
Binary files a/dataset/__pycache__/celeba.cpython-310.pyc and b/dataset/__pycache__/celeba.cpython-310.pyc differ
|
|
|
models/__pycache__/blocks.cpython-310.pyc
CHANGED
|
Binary files a/models/__pycache__/blocks.cpython-310.pyc and b/models/__pycache__/blocks.cpython-310.pyc differ
|
|
|
models/__pycache__/discriminator.cpython-310.pyc
CHANGED
|
Binary files a/models/__pycache__/discriminator.cpython-310.pyc and b/models/__pycache__/discriminator.cpython-310.pyc differ
|
|
|
models/__pycache__/lpips.cpython-310.pyc
CHANGED
|
Binary files a/models/__pycache__/lpips.cpython-310.pyc and b/models/__pycache__/lpips.cpython-310.pyc differ
|
|
|
models/__pycache__/vqvae.cpython-310.pyc
CHANGED
|
Binary files a/models/__pycache__/vqvae.cpython-310.pyc and b/models/__pycache__/vqvae.cpython-310.pyc differ
|
|
|
models/vqvae.py
CHANGED
|
@@ -152,9 +152,6 @@ class VQVAE(nn.Module):
|
|
| 152 |
return out
|
| 153 |
|
| 154 |
def forward(self, x):
|
| 155 |
-
print("Encoding")
|
| 156 |
z, quant_losses = self.encode(x)
|
| 157 |
-
print(f"z: {z},quant_losses: {quant_losses}")
|
| 158 |
out = self.decode(z)
|
| 159 |
-
print("Decoding")
|
| 160 |
return out, z, quant_losses
|
|
|
|
| 152 |
return out
|
| 153 |
|
| 154 |
def forward(self, x):
|
|
|
|
| 155 |
z, quant_losses = self.encode(x)
|
|
|
|
| 156 |
out = self.decode(z)
|
|
|
|
| 157 |
return out, z, quant_losses
|