Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- root@c0mp1lr:/root# python tf-vgg16.py
- 2018-04-18 19:57:46.084614: W tensorflow/stream_executor/rocm/rocm_driver.cc:405] creating context when one is currently active; existing: 0x7fd972a57460
- 2018-04-18 19:57:46.084734: I tensorflow/core/common_runtime/gpu/gpu_device.cc:907] Found device 0 with properties:
- name: Device 6863
- AMDGPU ISA: gfx900
- memoryClockRate (GHz) 1.6
- pciBusID 0000:2a:00.0
- Total memory: 15.98GiB
- Free memory: 15.73GiB
- 2018-04-18 19:57:46.084773: W tensorflow/stream_executor/rocm/rocm_driver.cc:405] creating context when one is currently active; existing: 0x7fd972a5e150
- 2018-04-18 19:57:46.084810: I tensorflow/core/common_runtime/gpu/gpu_device.cc:907] Found device 1 with properties:
- name: Device 687f
- AMDGPU ISA: gfx900
- memoryClockRate (GHz) 1.63
- pciBusID 0000:2d:00.0
- Total memory: 7.98GiB
- Free memory: 7.73GiB
- 2018-04-18 19:57:46.084834: I tensorflow/core/common_runtime/gpu/gpu_device.cc:801] Peer access not supported between device ordinals 0 and 1
- 2018-04-18 19:57:46.084842: I tensorflow/core/common_runtime/gpu/gpu_device.cc:801] Peer access not supported between device ordinals 1 and 0
- 2018-04-18 19:57:46.084851: I tensorflow/core/common_runtime/gpu/gpu_device.cc:929] DMA: 0 1
- 2018-04-18 19:57:46.084857: I tensorflow/core/common_runtime/gpu/gpu_device.cc:939] 0: Y N
- 2018-04-18 19:57:46.084864: I tensorflow/core/common_runtime/gpu/gpu_device.cc:939] 1: N Y
- 2018-04-18 19:57:46.084872: I tensorflow/core/common_runtime/gpu/gpu_device.cc:997] Creating TensorFlow device (/gpu:0) -> (device: 0, name: Device 6863, pci bus id: 0000:2a:00.0)
- 2018-04-18 19:57:46.084878: I tensorflow/core/common_runtime/gpu/gpu_device.cc:997] Creating TensorFlow device (/gpu:1) -> (device: 1, name: Device 687f, pci bus id: 0000:2d:00.0)
- Traceback (most recent call last):
- File "tf-vgg16.py", line 13, in <module>
- output = vgg16(input_img)
- File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/keras/python/keras/engine/topology.py", line 396, in __call__
- output = super(Layer, self).__call__(inputs, **kwargs)
- File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/layers/base.py", line 450, in __call__
- outputs = self.call(inputs, *args, **kwargs)
- File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/keras/python/keras/engine/topology.py", line 1742, in call
- output_tensors, _, _ = self.run_internal_graph(inputs, masks)
- File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/keras/python/keras/engine/topology.py", line 1922, in run_internal_graph
- output_tensors = _to_list(layer.call(computed_tensor, **kwargs))
- File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/layers/convolutional.py", line 158, in call
- data_format=utils.convert_data_format(self.data_format, self.rank + 2))
- File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/nn_ops.py", line 652, in convolution
- num_spatial_dims]))
- ValueError: number of input channels does not match corresponding dimension of filter, 224 != 3
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement