Hi,
I’m trying to train multi-agent reinforcement learning. To do so, each agent has distributed (separate) network model. Therefore, the distributed model has a number of nn.Module for each separate model.
I want to save the entire networks’ parameters to evaluate the trained model. How can I save the entire parameters?
As I know, using

d=model.state_dict()

and

torch.save(d, path)

is appropriate. But the model used in the above command seems to be linked for only one network model (not entire separate model).
How can I save all separate model?
Thank you