我知道'image_dataset_from_directory'可以用于tf nighty,但我正在使用GPU来训练我的数据,所以我安装了tf nightly GPU,它似乎不支持'image_data set_from_directory'?
有没有办法将"image_dataset_from_directory"与gpu版本的TF一起使用?
我解决了它。
而不是使用";从tensorflow.keras.preprocessing.image导入image_dataset_from_directory";使用";从tensorflow.keras.preprocessing import image_dataset_from_directory";