自定义nn.module层的注意事项

自定义nn.module层的注意事项

待办

使用张量构建参数,为什么不用require_grad=true,而是加到nn.Parameter中
使用张量构建参数,为什么不用require_grad=true,而是加到nn.Parameter中

要使用优化器优化要使用nn.parameter这个类,所有的参数都在这里边,所以要把参数都加进去
这样自定义的MyLinear和 nn.Linear一模一样

posted @ 2020-02-29 15:38  程序员成长  阅读(530)  评论(0)    收藏  举报