Download this file
1 2 3 4 5 6
from procedures.trainer import * print("Training CT-GAN Remover...") CTGAN_rem = Trainer(isInjector = False) CTGAN_rem.train(epochs=200, batch_size=32, sample_interval=50) print('Done.')