Spaces:
Running
on
T4
Running
on
T4
Update Modules/ControllabilityGAN/wgan/resnet_1.py
Browse files
Modules/ControllabilityGAN/wgan/resnet_1.py
CHANGED
|
@@ -76,7 +76,7 @@ class ResNet_G(nn.Module):
|
|
| 76 |
return out, l_1
|
| 77 |
return out
|
| 78 |
|
| 79 |
-
def sample_latent(self, n_samples, z_size, temperature
|
| 80 |
return torch.randn((n_samples, z_size)) * temperature
|
| 81 |
|
| 82 |
|
|
|
|
| 76 |
return out, l_1
|
| 77 |
return out
|
| 78 |
|
| 79 |
+
def sample_latent(self, n_samples, z_size, temperature):
|
| 80 |
return torch.randn((n_samples, z_size)) * temperature
|
| 81 |
|
| 82 |
|