def __init__(self, latent_size: int):
super().__init__()
self.__sequential_blocks = [
LinearGeneratorBlock(latent_size, 256),
LinearGeneratorBlock(256, 256),
LinearGeneratorBlock(256, 256),
nn.Linear(256, 28 * 28),
nn.Sigmoid(),
View(-1, 1, 28, 28)
]
self.main = nn.Sequential(*self.__sequential_blocks)
def __init__(self, latent_size: int):
super().__init__()
self.__sequential_blocks = [
nn.Linear(latent_size, 1024 * 8 * 8),
View(-1, 1024, 8, 8),
CnnDecoderBlock(1024, 512),
CnnDecoderBlock(512, 256),
CnnDecoderBlock(256, 128),
nn.ConvTranspose2d(128, 3, 3, 1, 1),
nn.Tanh(),
MapTanhZeroOne()
]
self.main = nn.Sequential(*self.__sequential_blocks)
def __init__(self, latent_size: int):
super().__init__()
self.__sequential_blocks = [
nn.Linear(latent_size, 256 * 7 * 7),
nn.ReLU(True),
View(-1, 256, 7, 7),
CnnDecoderBlock(256, 128),
CnnDecoderBlock(128, 64),
nn.ConvTranspose2d(64, 3, 3, 1, 1),
nn.Tanh(),
MapTanhZeroOne()
]
self.main = nn.Sequential(*self.__sequential_blocks)
def __init__(self, noise_dim: int):
super().__init__()
self.__sequential_blocks = [
LinearGeneratorBlock(noise_dim, 512),
LinearGeneratorBlock(512, 512),
LinearGeneratorBlock(512, 512),
LinearGeneratorBlock(512, 512),
LinearGeneratorBlock(512, 512),
LinearGeneratorBlock(512, 512),
nn.Linear(512, 28 * 28),
nn.Sigmoid(),
View(-1, 1, 28, 28)
]
self.main = nn.Sequential(*self.__sequential_blocks)
def __init__(self, noise_dim: int):
super().__init__()
self.__sequential_blocks = [
LinearGeneratorBlock(noise_dim, 128),
LinearGeneratorBlock(128, 128),
LinearGeneratorBlock(128, 128),
LinearGeneratorBlock(128, 256 * 8 * 8),
View(-1, 256, 8, 8),
CnnDecoderBlock(256, 128),
CnnDecoderBlock(128, 64),
nn.ConvTranspose2d(64, 3, 3, 1, 1),
nn.Tanh(),
MapTanhZeroOne()
]
self.main = nn.Sequential(*self.__sequential_blocks)
def setUp(self):
"""Set up tests."""
initLocale()
controller = MagicMock()
controller.getLifemapSize = MagicMock(return_value=(50, 50))
ruleSetup = (
'GameOfLIfe.txt',
0,
Colors(['White', 'Black']),
['Moore', '0', ''],
{
(0, (5, 3)): 1,
(1, (5, 3)): 1,
(1, (6, 2)): 1
},
['White', 'Black'],
)
controller.getRuleFiles = MagicMock(return_value=[ruleSetup])
self.view = View(controller)
def initView(controller):
"""Init view instance."""
return View(controller)
from modules.view import View from modules.model import Model view = View(10) # create view - 10 is field size model = Model(view) # add logic view.render()
