I am trying to use one of torchvision’s pretrained VGG models with torch.legacy.nn
, but I am running into issues with layers like Conv2d
, which are not supported by torch.legacy.nn
.
Or would it make more sense to load the model and then convert/change the problematic layers to their legacy.nn
equivalents? Would that even work?