The parameter was added to repo less than a month ago, and hasn’t been included in a release yet: https://github.com/pytorch/vision/commit/59858699563964e95931fd9328d88b67a7704587#diff-549cc9d2872b8748db818d9a51e34708 43. You can remove the 0.5 for now.