Python get_init_fn_for_scaffold Beispiele

Beispiel #1

Datei: train_textboxes_plusplus.py Projekt: lloydnguyen96/ML_TextBoxesPlusPlus

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)

Beispiel #2

Datei: train_ssd.py Projekt: wawaa/SSD.TensorFlow

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'
                                              })

Beispiel #3

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'
                                              })