Solved by ptrblck in post #2 
              
              
                You could create a method to load your weights. 
I created a small example for you. 
You could definitely write the code in a more compact way, so this should be a starter only.  
class SSD(nn.Module):
    def __init__(self, init_weights=True):
        super(SSD, self).__init__()

        # ========…