Python Helper Beispiele

Beispiel #1

import sys
sys.path.append('../../')

from tframe.utils.script_helper import Helper
from tframe.trainers.trainer import TrainerHub as Config

Helper.register_flags(Config)
s = Helper()
s.register('job-dir', '.')
# -----------------------------------------------------------------------------
# Configure data set here
# -----------------------------------------------------------------------------
s.register('max_iterations', 50000)

s.register('bits', 3)
s.register('sequence_length', 100)
# -----------------------------------------------------------------------------
# Specify summary file name and GPU ID here
# -----------------------------------------------------------------------------
summ_name = s.default_summ_name
gpu_id = 0

s.register('gather_summ_name', summ_name + '.sum')
s.register('gpu_id', gpu_id)
# -----------------------------------------------------------------------------
# Set up your models and run
# -----------------------------------------------------------------------------
s.register('state_size', 67)
s.register('lr', 0.001)

s.run(50)

Beispiel #2

Datei: s50_gam_rhn.py Projekt: samyeh0527/gam_rhn

import sys

sys.path.append('../../')

from tframe.utils.script_helper import Helper
from tframe.trainers.trainer import TrainerHub as Config

Helper.register_flags(Config)
s = Helper()
# -----------------------------------------------------------------------------
# Configure data set here
# -----------------------------------------------------------------------------
s.register('epoch', 1000)
# -----------------------------------------------------------------------------
# Specify summary file name and GPU ID here
# -----------------------------------------------------------------------------
gpu_id = 0
summ_name = s.default_summ_name

s.register('gather_summ_name', summ_name + '.sum')
s.register('gpu_id', gpu_id)
# -----------------------------------------------------------------------------
# Set up your models and run
# -----------------------------------------------------------------------------
s.register('lr', 0.0004, 0.0002, 0.0001, 0.00004, 0.00002, 0.00001)

s.run(1)

Beispiel #3

import sys
sys.path.append('../../')

from tframe.utils.script_helper import Helper
from tframe.trainers.trainer import TrainerHub as Config


Helper.register_flags(Config)
s = Helper()
# -----------------------------------------------------------------------------
# Configure data set here
# -----------------------------------------------------------------------------
s.register('max_iterations', 50000)

s.register('bits', 3)
s.register('sequence_length', 100)
# -----------------------------------------------------------------------------
# Specify summary file name and GPU ID here
# -----------------------------------------------------------------------------
summ_name = s.default_summ_name
gpu_id = 0

s.register('gather_summ_name', summ_name + '.sum')
s.register('gpu_id', gpu_id)
# -----------------------------------------------------------------------------
# Set up your models and run
# -----------------------------------------------------------------------------
s.register('gam_config', '6x10')
s.register('head_size', 20)
# s.register('sog_version', 0)

Beispiel #4

Datei: s01_mlp.py Projekt: winkywow/tframe

import sys
sys.path.append('../')
sys.path.append('../../')

from tframe.utils.script_helper import Helper

s = Helper()
# -----------------------------------------------------------------------------
# Configure data set here
# -----------------------------------------------------------------------------
s.register('epoch', 5)
# -----------------------------------------------------------------------------
# Specify summary file name and GPU ID here
# -----------------------------------------------------------------------------
gpu_id = 0
summ_name = s.default_summ_name

s.register('gather_summ_name', summ_name + '.sum')
s.register('gpu_id', gpu_id)
# -----------------------------------------------------------------------------
# Specified hyper-parameters
# -----------------------------------------------------------------------------
s.register('batch_size', 32, 64)

# -----------------------------------------------------------------------------
# Set up your models and run
# -----------------------------------------------------------------------------
s.register('gpu_memory_fraction', 0.3)
s.run(5, save=False)

Beispiel #5

Datei: s5_fsrnn.py Projekt: samyeh0527/gam_rhn

import sys
sys.path.append('../../')

from tframe.utils.script_helper import Helper
from tframe.trainers.trainer import TrainerHub as Config

Helper.register_flags(Config)
s = Helper()
s.register('job-dir', '.')
# -----------------------------------------------------------------------------
# Configure data set here
# -----------------------------------------------------------------------------
s.register('max_iterations', 50000)

s.register('bits', 3)
s.register('sequence_length', 100)
# -----------------------------------------------------------------------------
# Specify summary file name and GPU ID here
# -----------------------------------------------------------------------------
summ_name = s.default_summ_name
gpu_id = 0

s.register('gather_summ_name', summ_name + '.sum')
s.register('gpu_id', gpu_id)
# -----------------------------------------------------------------------------
# Set up your models and run
# -----------------------------------------------------------------------------
s.register('hyper_kernel', 'lstm')
s.register('fast_layers', 2, 3, 4, 5, 6)
s.register('lr', 0.001)

Beispiel #6

import sys

sys.path.append('../../')

from tframe.utils.script_helper import Helper
from tframe.trainers.trainer import TrainerHub as Config

Helper.register_flags(Config)
s = Helper()
# s.set_python_cmd_suffix()
# -----------------------------------------------------------------------------
# Configure data set here
# -----------------------------------------------------------------------------
s.register('epoch', 10000)
# -----------------------------------------------------------------------------
# Specify summary file name and GPU ID here
# -----------------------------------------------------------------------------
gpu_id = 0

summ_name = s.default_summ_name

s.register('gather_summ_name', summ_name + '.sum')
s.register('gpu_id', gpu_id)
# -----------------------------------------------------------------------------
# Set up your models and run
# -----------------------------------------------------------------------------
s.register('rhn_dropout', 0.1, 0.2, 0.3)
s.register('gam_dropout', 0.1, 0.2)
s.register('output_dropout', 0.1, 0.2, 0.3)
s.register('lr', 0.0008, 0.0004, 0.0001, 0.00002)
s.run(1)