search

gustavla / self-supervision

Training and evaluating self-supervised deep neural networks
BSD 3-Clause "New" or "Revised" License
29 stars 6 forks source link

Tensorflow code not responding #4

Open abhaymittal opened 6 years ago

abhaymittal commented 6 years ago

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