search

tensorlayer / SRGAN

Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network
https://github.com/tensorlayer/tensorlayerx
3.24k stars 813 forks source link

Error in restore_model #262

Open gralliry opened 1 month ago

gralliry commented 1 month ago 
Traceback (most recent call last):
  File "D:\Code\Python\SRGAN\train.py", line 46, in <module>
    VGG = vgg.VGG19(pretrained=True, end_with='pool4', mode='dynamic')
  File "D:\Code\Python\SRGAN\vgg.py", line 263, in vgg19
    restore_model(model, layer_type='vgg19')
  File "D:\Code\Python\SRGAN\vgg.py", line 174, in restore_model
    logging.info("  Loading %s in %s" % (str(val[1][0].shape), val[0]))
IndexError: too many indices for array: array is 0-dimensional, but 1 were indexed