[f84ece]: / 2B_train_remover.py

Download this file

6 lines (5 with data), 187 Bytes

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.')