hx173149 / C3D-tensorflow

C3D is a modified version of BVLC tensorflow to support 3D ConvNets.
MIT License
588 stars 262 forks source link

var_list = [v for v in tf.trainable_variables() if v.name.find("conv") != -1] #92

Open 953585130 opened 6 years ago

953585130 commented 6 years ago

I tried to modify the code to fine-tune (my own dataset), but it always failed. How should I modify it?

saver = tf.train.Saver(weights.values() + biases.values()) init = tf.global_variables_initializer()

Create a session for running Ops on the Graph.

sess = tf.Session( config=tf.ConfigProto(allow_soft_placement=True) ) sess.run(init) if os.path.isfile(model_filename) and use_pretrained_model: saver.restore(sess, model_filename)

please help!