no of epochs and learning rate
#1
by SampadKar - opened
for how many epochs you trained the model and with what learning rate ??
the model was trained with differential learning rates, lr=1e-4 for newly initialized layers 2e-5 for middle layers and 1e-5 for the backbone, I set epochs to 80 but I used early stopping so don't remember how many epochs in total were trained, you can find more details on the github : https://github.com/mohamedkhayat/transformers-for-fruit-object-detection-internship
if you're looking to use a model for inference I suggest using one of the rt detr v1 models like this one : https://huggingface.co/MohamedKhayat/fruit-detector-rtdetrv1-101