def train(self, epochs, batch_size=128, save_interval=50): # Load the dataset # pdb.set_trace() X_train = np.load(\'areadata_cGANvf.npy\')