a.amiri
/
GraphRNN2
Watch 1
Star 0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
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.
6 Commits
1 Branch
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
GraphRNN2/main_baselines/graphvae/Main_VAE_Args.py

Main_VAE_Args.py 345B
Raw Permalink Blame History

12345678910 
  1. 

  2. ### program configuration

  3. class Main_VAE_Args():

  4.     def __init__(self):

  5.         ### if clean tensorboard

  6.         self.load_epoch = 16

  7.         self.graph_type = 'IMDBBINARY'

  8.         self.dir_input = './'

  9.         self.model_save_path = self.dir_input+'model_save/' # only for nll evaluation

  10.         ### Which CUDA GPU device is used for training