Tensorflow 1.15 manual warm start

manual_warm_start.py

warm_starts = sorted(glob.glob(args.warm_start_from)) if args.warm_start_from is not None else None
warm_start_from = warm_starts[-1] if (warm_starts is not None and len(warm_starts)) else None
warm_start_from = os.path.join(warm_start_from, 'variables/variables')

id_assignment_map = {}
for myvar in tf.contrib.framework.get_variables_to_restore():
    myscope = os.path.dirname(myvar.name)
    if len(myscope):
        myscope += '/'
        id_assignment_map[myscope] = myscope
        
tf.train.init_from_checkpoint(warm_start_from, id_assignment_map)