George3d6:

ers


Didn’t u already have a non-loop version in your first post?
Sadly, not all loops can be replaced by Pytorch methods. In this case, the loops is fine because u r only iterating through a list of models, not using iteration inside the algorithm.
Also, if ur embedding_networks is a sequence of models, I guess u should change this line
for i, network in embedding_networks:

to
for i, network in enumerate(embedding_networks):

.