If I want to change the more complex network to do (ex: resnet), is it just a matter of changing the last fully connected layer to two fully connected layers to output the regression and classification results?