hi 各位巨大佬,之前用tf2,直接用继承model的方法创建的model很容易出现这个bug,不知道哪里有鬼,而参数我已经在init进行了初始化,按道理说不应该啊。卧槽,这是啥原因呢?
gradients = tape.gradient(loss, model.trainable_variables)
/data/prod/anaconda3/envs/xulm1/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer.py:1972 trainable_variables
return self.trainable_weights
/data/prod/anaconda3/envs/xulm1/lib/python3.8/site-packages/tensorflow/python/keras/engine/network.py:569 trainable_weights
trackable_layer_utils.gather_trainable_weights(
/data/prod/anaconda3/envs/xulm1/lib/python3.8/site-packages/tensorflow/python/training/tracking/layer_utils.py:271 gather_trainable_weights
weights += layer.trainable_weights
/data/prod/anaconda3/envs/xulm1/lib/python3.8/site-packa