Тензорный слой не может загружать предварительно подготовленный вес - PullRequest
Новая
       18

Тензорный слой не может загружать предварительно подготовленный вес

1 голос
/ 14 января 2020

У меня проблема с загрузкой предварительно обученных параметров в тензорный слой. Я пытаюсь использовать srgan (https://github.com/tensorlayer/srgan) и его предварительно обученные параметры (https://github.com/tensorlayer/srgan/releases/tag/1.2.0)

Применимый код ниже 

G = get_G([1, None, None, 3])
load_params = tl.files.load_npz(path='', name='g_srgan.npz')
tl.files.assign_weights(load_params, 
#G.load_weights(os.path.join("g_srgan.npz"))
G.eval()

Я получил ошибку: 

Traceback (most recent call last):
  File "train.py", line 194, in <module>
    evaluate(session)
  File "train.py", line 156, in evaluate
    tl.files.assign_weights(load_params, G)
  File "/usr/local/lib/python3.6/dist-packages/tensorlayer/files/utils.py", line 2023, in assign_weights
    ops.append(network.all_weights[idx].assign(param))
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/ops/resource_variable_ops.py", line 819, in assign
    self._shape.assert_is_compatible_with(value_tensor.shape)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/tensor_shape.py", line 1110, in assert_is_compatible_with
    raise ValueError("Shapes %s and %s are incompatible" % (self, other))
ValueError: Shapes (1, 1, 1, 64) and (64,) are incompatible

Пожалуйста, сообщите мне решение. Спасибо !!

...