torch >= 1.8