If you load the pretrained parameters, you need to specify aux_logits=True (or just leave the default).
In that case, slice the output as suggested in the other post.
Which kind of error are you getting then?