notebook 3, code cell in evaluation model section

Re: notebook 3, code cell in evaluation model section

par George Ciubotariu,
Nombre de réponses : 0
Notebook was run in Google Colab for testing it.
Indeed, when using CUDA, there should be a ".cpu()" call.
Took notice and modifying the notebook. Thank you!