self-supervision
self-supervision copied to clipboard
Tensorflow code not responding
Hi @gustavla,
I tried running the Tensorflow code that you have uploaded using the following instructions that you mentioned:
import deepdish as dd
import selfsup.model.vgg16
data = dd.io.load('vgg16.caffemodel.h5')
x = tf.placeholder(tf.float32, shape=[1, 224, 224, 3], name='x')
phase_test = tf.placeholder(tf.bool, name='phase_test')
z = selfsup.model.vgg16.build_network(x, parameters=data, phase_test=phase_test)
The code gets stuck somewhere. I did a preliminary debug and it seems that it occurs on calls to vgg_conv()
in line 302 of vgg16.py. I ran it on a titanx gpu for 4 hours and it was not able to return from build_network()
. Can you please see if you can fix it?
Regards, Abhay