Lewis_Liu:

I switched to using Queue. But I cannot avoid firstly getting the net tensors from state_dict right?


I don’t have the full context here. Can you let the processing holding the state_dict be the writer to the queue?