GraphRNN2/main_baselines/graphvae/Main_VAE_Args.py

### program configuration
class Main_VAE_Args():
    def __init__(self):
        ### if clean tensorboard
        self.load_epoch = 16
        self.graph_type = 'IMDBBINARY'
        self.dir_input = './'
        self.model_save_path = self.dir_input+'model_save/' # only for nll evaluation
        ### Which CUDA GPU device is used for training