Copying nn.Modules without shared memory
lenny
no, what you need to do is to send them model to the new process, and then do: <SCODE>import copy
model = copy.deepcopy(model)
<ECODE> share_memory() only shares the memory ahead of time (in case you want to reuse the shared memory for example)
smth
apaszke
Then could you give me an example of using load_state_dict() to copy a module?
Juna
He is referring to saving the state_dict (for instance into a file), sending it over to another process, then loading it with load_state_dict
JosueCom