nikigoli's picture
Tried commenting out model.cpu() part of tutorial count function
b812b2c verified