SRGAN
SRGAN copied to clipboard
AttributeError: 'Model' object has no attribute 'weights'
When I train the model with the dataset specified in the readme file. I got the following error
Traceback (most recent call last):
File ".\train.py", line 357, in <module>
train()
File ".\train.py", line 102, in train
grad = tape.gradient(mse_loss, G.weights)
AttributeError: 'Model' object has no attribute 'weights'
Running on Tensorflow 2.0.0 alpha0 Any idea how to fix it?
It seams to work after I change G.weights to G.trainable_weights
Same problem here. There are a few instances that need changing.
Same problem
It seams to work after I change
G.weightstoG.trainable_weights
It works perfect for me.
I update the version of tensorlayer > 2.0.0, and it works.