Hi, I'm trying to use your transfer learning scripts, but it seems that the explicit model construction using your nn_layers module creates a slightly different graph than the one that is necessary to load the pre-trained model.
NotFoundError (see above for traceback): Key conv_1_batch_norm/conv_1/batch_norm/conv_1_batch_norm/moments/Squeeze/ExponentialMovingAverage not found in checkpoint
The closest namespace in the pre-trained model that seems to be an analog to the node called out in the error message above is: conv_1_batch_norm/conv_1/batch_norm/conv_1_batch_norm/moments/moments_1/mean/ExponentialMovingAverage (DT_FLOAT) [64]
I think I narrowed the issue down to the batch normalization step in nn_layers, but I'm not able to figure out why the new graph construction doesn't match the pre-trained model. Any help would be greatly appreciated.
Thanks!
Hi, I'm trying to use your transfer learning scripts, but it seems that the explicit model construction using your nn_layers module creates a slightly different graph than the one that is necessary to load the pre-trained model.
NotFoundError (see above for traceback): Key conv_1_batch_norm/conv_1/batch_norm/conv_1_batch_norm/moments/Squeeze/ExponentialMovingAverage not found in checkpointThe closest namespace in the pre-trained model that seems to be an analog to the node called out in the error message above is: conv_1_batch_norm/conv_1/batch_norm/conv_1_batch_norm/moments/moments_1/mean/ExponentialMovingAverage (DT_FLOAT) [64]
I think I narrowed the issue down to the batch normalization step in nn_layers, but I'm not able to figure out why the new graph construction doesn't match the pre-trained model. Any help would be greatly appreciated.
Thanks!