CUDA error: CUBLAS_STATUS_ALLOC_FAILED when running loss.backward()

前端 未结 0 492
悲&欢浪女
悲&欢浪女 2021-02-05 10:00

I have a simple model, and I\'m trying to run a classification task.

The following is my model

num_epochs = 10
batch_size = 64
learning_rate = 0.001

input         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题