硬件 GPU
软件:pycharm+pytorch+python2
参考文献:
https://blog.csdn.net/tsq292978891/article/details/79364140
这一篇在于关注如何构建module
https://www.cnblogs.com/hellcat/p/8496850.html
这一篇主要关注最后的数据传入,先进行初始化,在传入参数。(对自己有启发)
https://www.cnblogs.com/hellcat/p/8473412.html
这一篇主要关注在自定义module时候各个部分的意义以及注意事项。(理论)
感谢以上三篇的作者!