Closed 15966697671 closed 3 years ago
Sorry for the latter response. First, you should build up a data loader to read and process your data. Second, you modify the output channel of the last convolutional layer of ct_head to the number of categories.
I want to perform transfer learning to train the model with the pretrained weights on my own small image dataset. My small dataset category is completely different from that of Hoi. Let me know how to go about it.thank you.