import tensorflow as tf

        optimizer_vars = set([var.name for var in optimizer.variables()])
        prefix = 'resnet%s/' % params['resnet_depth']
        resnet_vars = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, prefix)
        vars_to_load = {}
