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 |
|