Card

Git

Discussions

1
2
3
4
5
6

from procedures.trainer import *

print("Training CT-GAN Injector...")
CTGAN_inj = Trainer(isInjector = True)
CTGAN_inj.train(epochs=200, batch_size=32, sample_interval=50)
print('Done.')