search

cent664 / SSRIW

Tensorflow implementation of 'Robust Image Watermarking based on Cross-Attention and Invariant Domain Learning'
https://arxiv.org/abs/2310.05395
8 stars 0 forks source link

runtime error：'tensorflow.python.framework.ops.EagerTensor' object has no attribute 'values' #2

Open shiboliu opened 1 month ago

shiboliu commented 1 month ago

截图 2024-03-22 11-32-37 截图 2024-03-22 11-33-13

why？？？

cent664 commented 1 month ago

I used distributed training across multiple GPUs and TPUs to train the model. The iterator produces distributed datasets. I used .values to access them.

https://www.tensorflow.org/api_docs/python/tf/distribute/DistributedValues