class DQNnnModule def __init__self in_planes outputs stride=1 hidden_size=128 superDQN self__init__ # selfconv1 = nnConv2din_channels=in_planes out_channels=256 kernel_size=3 #
可以像下面这样调用该网络结构:
dqn = DQN(in_planes=3, outputs=4) 输入参数in_planes表示输入图像的通道数,outputs表示输出的动作数。创建实例后,可以使用该网络进行前向传播和反向传播等操作。
原文地址: https://www.cveoy.top/t/topic/bC0f 著作权归作者所有。请勿转载和采集!