遇到一个奇怪的问题
运行上面的代码,导致kernel直接重启,如下图,而且没有报错
将arange(0, 12)改成浮点型arange(0., 12.)可以正常Dropout,输出结果。
或者写成:
input = t.arange(0, 12).float().view(3, 4)
添加一个.float(),也没问题。
目前只在Dropout上发现这个问题,其他方法没试过。
遇到一个奇怪的问题
运行上面的代码,导致kernel直接重启,如下图,而且没有报错
将arange(0, 12)改成浮点型arange(0., 12.)可以正常Dropout,输出结果。
或者写成:
input = t.arange(0, 12).float().view(3, 4)
添加一个.float(),也没问题。
目前只在Dropout上发现这个问题,其他方法没试过。