Hi, i’m a newbie Pytorch.
Now, I have a problem with my work. I dont find any solution for it.
I want to implement one model like behind image. It will run on single GPU.
My inputs are extended more one dimensions (5 dimensions instead of 4 ). In models, there is 2 submodels. Each submodel has one inputs like my image.
I have to implement 2 submodels which run parallelly to improve performance.
I try to use multhreading on one GPU but the time is the same with sequential models.
Thanks for helping.

t.png700×300 12.7 KB