import os


class DatasetConfig:
    def __init__(self):
        self.project_data_root = './datasets'
        self.mvtecPath = './data/mvtec'
        if not os.path.exists(self.mvtecPath):
            print('MvTec dataset path is wrong')
            exit(1)
        self.mnistPath = './data/mnist'
        self.cifar10Path = './data/cifar10'

        self.shanghaitechPath = './data/mSTC'


class TrainingConfig:
    def __init__(self):
        self.dataset_name = 'mvtec'
        self.batch_size = 4

        self.img_size = 256

        self.epoch = 300
        self.optim_lr = 0.5

        self.eval_span = 5
        self.train_log_span = 100
        self.save_span = 100

        self.ckpt_path = os.path.join('checkpoints',self.dataset_name)