def train_once_(sess, step, is_start=False, fixed_step=None,
num_epochs=None, model_path=None, use_horovod=False):
train_once(sess,
step,
ops,
names,
gen_feed_dict_fn,
deal_results_fn,
interval_steps,
eval_ops,
eval_names,
gen_eval_feed_dict_fn,
deal_eval_results_fn,
valid_interval_steps,
print_time,
print_avg_loss,
model_dir,
log_dir,
is_start,
num_steps_per_epoch,
metric_eval_fn=metric_eval_fn,
metric_eval_interval_steps=metric_eval_interval_steps,
summary_excls=summary_excls,
fixed_step=fixed_step,
eval_loops=eval_loops,
learning_rate=learning_rate,
learning_rate_patience=learning_rate_patience,
learning_rate_decay_factor=learning_rate_decay_factor,
num_epochs=num_epochs,
model_path=model_path,
use_horovod=use_horovod
)
def train_once_(sess, step):
train_once(sess,
step,
ops,
names,
gen_feed_dict_fn,
deal_results_fn,
interval_steps,
eval_ops,
eval_names,
gen_eval_feed_dict_fn,
deal_eval_results_fn,
valid_interval_steps,
print_time,
print_avg_loss,
log_dir=log_dir,
num_steps_per_epoch=num_steps_per_epoch,
metric_eval_fn=metric_eval_fn,
metric_eval_interval_steps=metric_eval_interval_steps)
def train_once_(sess, step, is_start=False):
train_once(sess,
step,
ops,
names,
gen_feed_dict,
deal_results,
interval_steps,
eval_ops,
eval_names,
gen_eval_feed_dict,
deal_eval_results,
eval_interval_steps,
print_time,
print_avg_loss,
model_dir,
log_dir,
is_start,
num_steps_per_epoch,
metric_eval_function=metric_eval_function,
metric_eval_interval_steps=metric_eval_interval_steps)
def train_once_(sess, step, is_start=False, fixed_step=None):
train_once(sess,
step,
ops,
names,
gen_feed_dict_fn,
deal_results_fn,
interval_steps,
eval_ops,
eval_names,
gen_eval_feed_dict_fn,
deal_eval_results_fn,
eval_interval_steps,
print_time,
print_avg_loss,
model_dir,
log_dir,
is_start,
num_steps_per_epoch,
metric_eval_fn=metric_eval_fn,
metric_eval_interval_steps=metric_eval_interval_steps,
summary_excls=summary_excls,
fixed_step=fixed_step)