def run_test(cfile):
"""
Runs a configuration file
"""
config = load_config_file(f"../TestConfigs/{cfile}.json", id=True)
# print(config)
dispatch = TrainDispatch()
runconfig = RunConfig(impl=1,
verbose=True,
tboard=False,
best=True,
early=True,
multi=1,
proxy=False,
save=False,
remote=False,
info=True,
log=False)
train_process, architecture = dispatch.dispatch(config['arch']['mode'])
lresults = train_process(architecture, config, runconfig)
print(lresults)
:Description: Test
:Authors: bejar
:Version:
:Created on: 09/11/2018 11:52
"""
from Wind.Architectures.RNNS2SArchitecture import RNNS2SArchitecture
from Wind.Misc import load_config_file
from Wind.Train import RunConfig
__author__ = 'bejar'
if __name__ == '__main__':
config = load_config_file("../examples/config_RNN_ED_s2s.json", id=True)
config['idimensions'] = (12, 6)
config['odimensions'] = 12
runconfig = RunConfig(1, 0, False, False, False, False, False, False,
False)
print(config)
net = RNNS2SArchitecture(config, runconfig)
net.generate_model()
net.summary()
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# The GPU id to use
os.environ["CUDA_VISIBLE_DEVICES"] = args.dev
verbose = 1 if args.verbose else 0
impl = 2 if args.gpu else 1
if args.config is None:
config = getconfig(proxy=args.proxy, mode=args.exp, secpat=args.secpat)
elif args.mino:
config = load_config_file(args.config, id=False, mino=True)
else:
config = load_config_file(args.config, id=True)
run_config = RunConfig(impl, verbose, args.tboard, args.best, args.early,
args.multi, args.proxy, args.save, args.remote,
args.info, args.log)
dispatch = TrainDispatch()
if config is not None:
############################################
# Data
if not 'site' in config:
site = config['data']['datanames'][0].split('-')
config['site'] = f"{site[0]}-{site[1]}"
# try:
print(
f"Running job {config['_id']} {config['site']} {config['arch']['mode']} {strftime('%Y-%m-%d %H:%M:%S')}"