有没有人可以帮助我实现Pytorch的课程辍学。在此先感谢您,将提供任何帮助。
我想在Pytorch中进行一些课程辍学的实验。课程辍学尝试使用时间安排来调整神经网络中的辍学率。相关论文可以从here下载。 可以在here
中找到python中的源代码答案 0 :(得分:1)
我快速浏览了这篇论文,似乎主要的想法是使用计划的辍学率而不是固定的辍学率。
火炬已经有一个Dropout模块:torch.nn.modules.dropout.Dropout
。
对于使用辍学模块dropout
的自定义神经网络,您只需在优化步骤之间修改dropout.p
就可以安排辍学率。