def forward(self, img):
    '\n        Your code here\n        Predict the aim point in image coordinate, given the supertuxkart image\n        @img: (B,3,96,128)\n        return (B,2)\n        '
    x = self._conv(img)
    return spatial_argmax(x[:, 0])
Your code here
Predict the aim point in image coordinate, given the supertuxkart image
@img: (B,3,96,128)
return (B,2)
def forward(self, img):
    '\n        Your code here\n        Predict the aim point in image coordinate, given the supertuxkart image\n        @img: (B,3,96,128)\n        return (B,2)\n        '
    x = self._conv(img)
    return spatial_argmax(x[:, 0])