报错的函数调用位置
正确的函数调用位置
a模块调用b模块
理想状态 在a模块里定义调用b模块的函数,并在a模块里启动该函数。顺理成章
错误状态 在a模块外定义调用b模块的函数,并在a模块里启动该函数。造成gpu设备的连接混乱
解决方案:在forward里面,定义调用函数,并启动
报错的函数调用位置
正确的函数调用位置
a模块调用b模块
理想状态 在a模块里定义调用b模块的函数,并在a模块里启动该函数。顺理成章
错误状态 在a模块外定义调用b模块的函数,并在a模块里启动该函数。造成gpu设备的连接混乱
解决方案:在forward里面,定义调用函数,并启动