Eduin Hernandez

You can set CUDA = int(False) in the run_[model].py.

I had the same issue. It turns out to be an issue with my dataset. To overcome this, I modified the following in utils/text_process.py: ``` def tokens_to_tensor(tokens, dictionary): tensor =...

Try [url](https://github.com/phibenz/double-targeted-uap.pytorch/blob/main/dataset_utils/ycb_downloader.py) instead.

The current version requires a downgrade of tensorflow and tensorflow-probability package: `!pip install --quiet tensorflow==2.8.0` `!pip install --quiet tensorflow-probability==0.15.0` `!pip install --quiet tensorflow-federated==0.20.0`