ZhaiJiaKai

> 我在训练中应用了 layers.py 中的损失函数，但我发现它不起作用。当我尝试用MNIST数据集训练一个简单的CNN时，我检查了数据集的formate和你的格式，我发现两者之间存在差异。 > > 假设有 4 个类，例如 0、1、2、3。我的网络的输出类似于 [0.1， 0.2， 0.3， 0.4]，目标类似于 [3]，但您的网络类似于 [[0， 0.1]、[1， 0.2]、[2， 0.3]、[3， 0.4]] 和 [[0， 0]、[1， 0]、[2， 0]、[3， 1]]。所以我做了一个转换，这是我的代码。 [loss.txt](https://github.com/dfdazac/wassdistance/files/9542249/loss.txt) Have you solved...

> 所以我找到了一个解决方法，也许其他人也有同样的问题，所以： > > [解决方法] > > * 卸载 Geomloss 和 Pykeops > * 在 /home/username/.cache 中删除 keops 缓存文件夹（风险自负） > * 安装 pykeops==2.0 和 geomloss > * 在您的 main 中，首先导入...