在Pytorch中实现课程辍学

时间:2018-08-23 01:03:43

标签: pytorch

有没有人可以帮助我实现Pytorch的课程辍学。在此先感谢您,将提供任何帮助。

我想在Pytorch中进行一些课程辍学的实验。课程辍学尝试使用时间安排来调整神经网络中的辍学率。相关论文可以从here下载。 可以在here

中找到python中的源代码

1 个答案:

答案 0 :(得分:1)

我快速浏览了这篇论文,似乎主要的想法是使用计划的辍学率而不是固定的辍学率。

火炬已经有一个Dropout模块:torch.nn.modules.dropout.Dropout

对于使用辍学模块dropout的自定义神经网络,您只需在优化步骤之间修改dropout.p就可以安排辍学率。