search

hx173149 / C3D-tensorflow

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

There are special function with varlist2& varlist1, grads1&grads2... #86

Open Ai-is-light opened 6 years ago

Ai-is-light commented 6 years ago 
   varlist2 = [ weights['out'],biases['out'] ]
    varlist1 = list( set(weights.values() + biases.values()) - set(varlist2) )
    logit = c3d_model.inference_c3d(
                    images_placeholder[gpu_index * FLAGS.batch_size:(gpu_index + 1) * FLAGS.batch_size,:,:,:,:],
                    0.5,
                    FLAGS.batch_size,
                    weights,
                    biases
                    )
    loss_name_scope = ('gpud_%d_loss' % gpu_index)
    loss = tower_loss(
                    loss_name_scope,
                    logit,
                    labels_placeholder[gpu_index * FLAGS.batch_size:(gpu_index + 1) * FLAGS.batch_size]
                    )
    grads1 = opt_stable.compute_gradients(loss, varlist1)
    grads2 = opt_finetuning.compute_gradients(loss, varlist2)
    tower_grads1.append(grads1)
    tower_grads2.append(grads2)