paul-corbalan
/
wasserstein-gan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Container -> Module (remove depreceated code)

This commit is contained in:
soumith 2017-02-27 12:39:45 -08:00
parent 93f26f9e4c
commit d9ad2bb847
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
models/dcgan.py
View File

@ -2,7 +2,7 @@ import torch
import torch.nn as nn
import torch.nn.parallel
class DCGAN_D(nn.Container):
class DCGAN_D(nn.Module):
def __init__(self, isize, nz, nc, ndf, ngpu, n_extra_layers=0):
super(DCGAN_D, self).__init__()
self.ngpu = ngpu
@ -51,7 +51,7 @@ class DCGAN_D(nn.Container):
output = output.mean(0)
return output.view(1)
class DCGAN_G(nn.Container):
class DCGAN_G(nn.Module):
def __init__(self, isize, nz, nc, ngf, ngpu, n_extra_layers=0):
super(DCGAN_G, self).__init__()
self.ngpu = ngpu
@ -104,7 +104,7 @@ class DCGAN_G(nn.Container):
return nn.parallel.data_parallel(self.main, input, gpu_ids)
###############################################################################
class DCGAN_D_nobn(nn.Container):
class DCGAN_D_nobn(nn.Module):
def __init__(self, isize, nz, nc, ndf, ngpu, n_extra_layers=0):
super(DCGAN_D_nobn, self).__init__()
self.ngpu = ngpu
@ -150,7 +150,7 @@ class DCGAN_D_nobn(nn.Container):
output = output.mean(0)
return output.view(1)
class DCGAN_G_nobn(nn.Container):
class DCGAN_G_nobn(nn.Module):
def __init__(self, isize, nz, nc, ngf, ngpu, n_extra_layers=0):
super(DCGAN_G_nobn, self).__init__()
self.ngpu = ngpu

4
models/mlp.py
View File

@ -5,7 +5,7 @@ from __future__ import unicode_literals
import torch
import torch.nn as nn
class MLP_G(nn.Container):
class MLP_G(nn.Module):
def __init__(self, isize, nz, nc, ngf, ngpu):
super(MLP_G, self).__init__()
self.ngpu = ngpu
@ -34,7 +34,7 @@ class MLP_G(nn.Container):
return out.view(out.size(0), self.nc, self.isize, self.isize)
class MLP_D(nn.Container):
class MLP_D(nn.Module):
def __init__(self, isize, nz, nc, ndf, ngpu):
super(MLP_D, self).__init__()
self.ngpu = ngpu