def get_init_fn(): return scaffolds.get_init_fn_for_scaffold( FLAGS.model_dir, # First FLAGS.checkpoint_path, # Second FLAGS.model_scope, FLAGS.checkpoint_model_scope, FLAGS.checkpoint_exclude_scopes, FLAGS.ignore_missing_vars, name_remap=None)
def get_init_fn(): return scaffolds.get_init_fn_for_scaffold(FLAGS.model_dir, FLAGS.checkpoint_path, FLAGS.model_scope, FLAGS.checkpoint_model_scope, FLAGS.checkpoint_exclude_scopes, FLAGS.ignore_missing_vars, name_remap={ '/kernel': '/weights', '/bias': '/biases' })
def get_init_fn(): # `scaffolds` is used to handle the model/checkpoints/reload related parameters return scaffolds.get_init_fn_for_scaffold(FLAGS.model_dir, FLAGS.checkpoint_path, FLAGS.model_scope, FLAGS.checkpoint_model_scope, FLAGS.checkpoint_exclude_scopes, FLAGS.ignore_missing_vars, name_remap={ '/kernel': '/weights', '/bias': '/biases' })