PyTorch Forums
Parallel execution of modules in nn.ModuleList
smell_well
(smell well)
May 24, 2022, 5:12pm
11
Hi!
Did you find a good method to deal with the problem? Use group conv is constrained.
How to speed up small parallel nn.Linear blocks (or small parallel matrix multiplications or group convolutions)?
show post in topic