loading pretrained model

[xllohh]

Hello, may I ask how the code you wrote trains your actual classification when loading the pre trained model? After reading the relevant code you wrote, it was not successfully modified. Can you tell me in which section to make the changes？