def train_main():
banner('Network Init')
nn = F2PEncoderDecoder(parser())
nn.identify_system()
# Bring in data:
ldrs = mixamo_loader_set(nn.hp)
# Supply the network with the loaders:
trainer = LightningTrainer(nn, ldrs)
trainer.train()
trainer.test()
trainer.finalize()
def test_main():
nn = F2PEncoderDecoder(parser())
ds = FullPartDatasetMenu.get('DFaustPyProj')
test_ldr = ds.loaders(s_nums=[1000, 1000], s_transform=[Center()], batch_size=nn.hp.batch_size,
device=nn.hp.dev, n_channels=nn.hp.in_channels, method='f2p')
trainer = LightningTrainer(nn, [None, None, test_ldr])
trainer.test()
trainer.finalize()