# Function def get_variables_to_restore(variables, var_keep_dic): variables_to_restore = [] for v in variables: # one can do include or exclude operations here. if v.name.split(':')[0] in var_keep_dic: print("Variables restored: %s" % v.name) variables_to_restore.append(v) return variables_to_restore # get the checkpoint of the old-trained model ckpt = tf.train.get_checkpoint_state(dir_of_ckpt) ..

!wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb !dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb !apt-key add /var/cuda-repo-8-0-local-ga2/7fa2af80.pub !apt-get update !apt-get install cuda=8.0.61-1 !apt autoremove