fix the runtime error

#7
by Zugo16 - opened

Hello there,
could you fix the runtime error please?
thanks.

pytorch org

Done :)

Sign up or log in to comment