weights/best_weight.pth