6e0c869f5e17c734010fe7cc1b975ef178af9d1c,agents/dagger/train/train.py,,run,#Any#Any#Any#,45

Before Change

        pretrained_var_map[v.op.name[6:]] = v

    alexnet_saver = tf.train.Saver(pretrained_var_map)

    def init_fn(ses):
        log.info("Initializing parameters.")
        if not has_stuff(c.BVLC_CKPT_PATH):

After Change

def run(resume_dir=None, data_dir=c.RECORDING_DIR, agent_name=None):
    // TODO: Clean up the net_name stuff
    if agent_name is None:
        net_name = net.ALEXNET_NAME
    elif agent_name == "dagger_mobilenet_v2":
        net_name = net.MOBILENET_V2_NAME
    else:
        raise NotImplementedError("%r agent_name not associated with trunk net" % agent_name)
    os.makedirs(c.TENSORFLOW_OUT_DIR, exist_ok=True)
    if resume_dir is not None:
        date_str = resume_dir[resume_dir.rindex("/") + 1:resume_dir.rindex("_")]
    else:

Instances