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