iter_4000.pth