Commit 6fa34967 authored by PhilFischer's avatar PhilFischer
Browse files

Set final model and experiment names

parent 5659f012
......@@ -71,7 +71,7 @@ if __name__ == '__main__':
# [-15921, -10503, 14.167, 14.045]
hparams = [
hp.HParam('architecture', hp.Discrete(['σ-VAE-2'])),
hp.HParam('architecture', hp.Discrete(['σ-VAE'])),
hp.HParam('likelihood', hp.Discrete(['sigma'])),
hp.HParam('output_activation', hp.Discrete(['linear'])),
hp.HParam('log_lr', hp.RealInterval(-4., -4.)),
......@@ -100,7 +100,7 @@ if __name__ == '__main__':
]
log_dir = 'logs'
ex_name = os.path.join('pex02', datetime.now().strftime("%y%m%d-%H%M%S"))
ex_name = os.path.join('ex01', datetime.now().strftime("%y%m%d-%H%M%S"))
tuner = Tuner((train, s_train), (test, s_test), define_model, hparams, metrics, log_dir, oracle, seed=SEED)
tuner.tune(ex_name, runs=1, epochs=1000)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment