If “darknet weights” refer to a pretrained model, and you’ve fine-tuned it for 20 epochs, I would assume you want to load your fine-tuned weights instead. So yes, I guess you might want to use model.load_state_dict(your_fientuned_checkpoint), but it depends on your actual use case.