Pytorch 在 forward 函数中加速神经网络。
最近在研究EDSR代码(项目地址:https://github.com/sksq96/pytorch-summary)的时候看到了forward_chop function,该参数的help写的是parser.add_argument('--chop', action='store_true', help='use memory-efficient for...
原创
2020-02-27 02:54:15 ·
921 阅读 ·
0 评论