from training.consistency_trainers import *

from training.vanilla_trainer import *

import sys

if __name__ == '__main__':

    config = {"model": "InductiveNet",

              "device": "cuda",

              "lr": 0.00001,

              "batch_size": 4,

              "epochs": 250,

              "use_inpainter": False}

    trainer = EnsembleConsistencyTrainer("ensemble", config)

    trainer.train()