getting error while running test_network.py

[neerajkuhike]

Hi ,

I am getting the below error while running test_network.py

Traceback (most recent call last):
File "test_network.py", line 188, in
pred = get_results(m, dat)
File "test_network.py", line 55, in get_results
out = m([images, vertex_label, J_2d])
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/keras/engine/base_layer.py", line 679, in call
outputs = self.call(inputs, *args, **kwargs)
File "/home/neerajku/neerajAvatar/MultiGarmentNetwork/network/base_network.py", line 308, in call
lat_codes = [self.top_([q, j]) for q, j in zip(images, Js)]
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/keras/engine/base_layer.py", line 679, in call
outputs = self.call(inputs, *args, **kwargs)
File "/home/neerajku/neerajAvatar/MultiGarmentNetwork/network/base_network.py", line 153, in call
x = self.conv1(inp)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/keras/engine/base_layer.py", line 676, in call
self._maybe_build(inputs)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/keras/engine/base_layer.py", line 1881, in _maybe_build
self.build(input_shapes)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/keras/layers/convolutional.py", line 165, in build
dtype=self.dtype)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/keras/engine/base_layer.py", line 384, in add_weight
aggregation=aggregation)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/training/tracking/base.py", line 644, in _add_variable_with_custom_getter
name=name, shape=shape)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/training/tracking/base.py", line 711, in _preload_simple_restoration
checkpoint_position=checkpoint_position, shape=shape)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/training/tracking/base.py", line 71, in init
self.wrapped_value = checkpoint_position.value_tensors()[VARIABLE_VALUE_KEY]
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/training/tracking/base.py", line 306, in value_tensors
value_tensors[serialized_tensor.name] = array_ops.identity(value)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/util/dispatch.py", line 180, in wrapper
return target(*args, **kwargs)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/ops/array_ops.py", line 81, in identity
copied = input._copy() # pylint: disable=protected-access
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 884, in _copy
new_tensor = self._copy_nograd(ctx, device_name)
File "/home/neerajku/.virtualenv/pymesh2.7/local/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 877, in _copy_nograd
new_tensor = self._copy_to_device(context=ctx._handle, device=device_name)
RuntimeError: Error copying tensor to device: /job:localhost/replica:0/task:0/device:GPU:0. /job:localhost/replica:0/task:0/device:GPU:0 unknown device.