pytorch注意事项

1、定义网络时init和forward一块写

2、没有训练参数的层写到init中

3、注意调整网络的train和eval模式

posted @ 2020-08-03 11:06  6+0  阅读(162)  评论(0)  编辑  收藏  举报