# model path declarations
prefix = "him2017-100"
checkpoint_dir = r"/home/matthias-k/GraphLSTM_data/%s" % prefix
# number of timesteps to be simulated (each step, the same data is fed)
graphlstm_timesteps = 1
learning_rate = 1e-3
model_name = "dpren"
checkpoint_dir += r"/%s" % model_name
tensorboard_dir = checkpoint_dir + r"/tensorboard"
# load dataset
HIM2017 = dataset_loaders.HIM2017Loader(train_validate_split=1)
# # PREPARE SESSION
# set Keras session
config = tf.ConfigProto(allow_soft_placement=True)
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
K.set_session(sess)
# # BUILD MODEL
print("\n### Building Model: %s ###\n" % model_name)
print("Building DP+REN network …")
import os
prefix, model_name, epoch = get_prefix_model_name_optionally_epoch()
checkpoint_dir = r"/home/matthias-k/GraphLSTM_data/%s" % prefix
# number of timesteps to be simulated (each step, the same data is fed)
graphlstm_timesteps = 2
learning_rate = 1e-3
checkpoint_dir += r"/%s" % model_name
tensorboard_dir = checkpoint_dir + r"/tensorboard/test_him2017"
# load dataset
HIM2017 = dataset_loaders.HIM2017Loader()
# # PREPARE SESSION
config = tf.ConfigProto(allow_soft_placement=True)
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
K.set_session(sess)
# # LOAD MODEL
print("\n### Loading Model: %s ###\n" % model_name)
epoch_str = "Last epoch" if epoch is None else ("Epoch %i" % epoch)
print("## %s ##\n" % epoch_str)
input_shape = [None, *re.Const.MODEL_IMAGE_SHAPE]
