解决A3C训练中网络参数都为0的问题
问题:
在训练A3C的过程中, 本来想用cuda进行多进程训练, 即
import torch
import torch.multiprocessing as mp
global_net = Net()
global_net.cuda()
global_net.share_memory
class Worker(mp.Process):
def __init__(net):
super().__init__()
self.net = global_net
def run():
....
原创
2022-05-13 18:05:54 ·
509 阅读 ·
1 评论