Train the final layer

[monjurulkarim]

Hi, Thanks for this amazing repository.
I am trying to train only the final layer ( network head) with my own very small dataset. How can I do that? Can you please help?

[dbolya]

You can do this by following the fine tuning procedure (#36) and then here:

yolact/yolact.py

Line 628 in f54b0a5

p = pred_layer(pred_x)

replace that with

p = pred_layer(pred_x.detach())

[edrica-caruana]

Hi, I am trying to train on a very small dataset consisting of only 10 train images and 5 val images. When calculating mAP it is remaining everything 0. Is it because the dataset is relatively small? Thanks