import gpulearn_z_x
import sys

if 'svhn' in sys.argv[1]:
    gpulearn_z_x.main(dataset=sys.argv[1],
                      n_z=300,
                      n_hidden=(500, 500),
                      seed=0,
                      comment='',
                      gfx=True)
elif sys.argv[1] == 'mnist':
    n_hidden = (500, 500)
    if len(sys.argv) == 4:
        n_hidden = [int(sys.argv[2])] * int(sys.argv[3])
    gpulearn_z_x.main(dataset='mnist',
                      n_z=50,
                      n_hidden=n_hidden,
                      seed=0,
                      comment='',
                      gfx=True)

# gpulearn_z_x.main(n_data=50000, dataset='svhn_pca', n_z=300, n_hidden=(500,500), seed=0)
Esempio n. 2
0
'''
modified by Chongxuan Li ([email protected])
'''
import gpulearn_z_x
import sys, os
import time
import color

n_hidden = (500, 500)
if len(sys.argv) > 2:
    n_hidden = tuple([int(x) for x in sys.argv[2:]])
nz = 500
if os.environ.has_key('nz'):
    nz = int(os.environ['nz'])
if os.environ.has_key('random_seed'):
    seed = 0
    if bool(int(os.environ['random_seed'])):
        seed = int(time.time())
    color.printRed('random_seed ' + str(seed))
else:
    seed = int(time.time())
    color.printRed('random_seed ' + str(seed))
gpulearn_z_x.main(dataset=sys.argv[1],
                  n_z=nz,
                  n_hidden=n_hidden,
                  seed=seed,
                  comment='',
                  gfx=True)
Esempio n. 3
0
'''
modified by Chongxuan Li ([email protected])
'''
import gpulearn_z_x
import sys, os
import time
import color

n_hidden = (500,500)
if len(sys.argv) > 2:
  n_hidden = tuple([int(x) for x in sys.argv[2:]])
nz=500
if os.environ.has_key('nz'):
  nz = int(os.environ['nz'])
if os.environ.has_key('random_seed'):
    seed = 0
    if bool(int(os.environ['random_seed'])):
        seed = int(time.time())
    color.printRed('random_seed ' + str(seed))
else:
    seed = int(time.time())
    color.printRed('random_seed ' + str(seed))
gpulearn_z_x.main(dataset=sys.argv[1], n_z=nz, n_hidden=n_hidden, seed=seed, comment='', gfx=True)

Esempio n. 4
0
import gpulearn_z_x
import sys

if 'svhn' in sys.argv[1]:
    gpulearn_z_x.main(dataset=sys.argv[1], n_z=300, n_hidden=(500,500), seed=0, comment='', gfx=True)
elif sys.argv[1] == 'mnist':
    n_hidden = (500,500)
    if len(sys.argv) == 4:
        n_hidden = [int(sys.argv[2])] * int(sys.argv[3])
    gpulearn_z_x.main(dataset='mnist', n_z=50, n_hidden=n_hidden, seed=0, comment='', gfx=True)

#gpulearn_z_x.main(n_data=50000, dataset='svhn_pca', n_z=300, n_hidden=(500,500), seed=0)