You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Main_VAE_Args.py 345B

4 years ago
12345678910
  1. ### program configuration
  2. class Main_VAE_Args():
  3. def __init__(self):
  4. ### if clean tensorboard
  5. self.load_epoch = 16
  6. self.graph_type = 'IMDBBINARY'
  7. self.dir_input = './'
  8. self.model_save_path = self.dir_input+'model_save/' # only for nll evaluation
  9. ### Which CUDA GPU device is used for training