去掉不必要的cuda占用
一般将模型,输入变量和标签使用cuda,在模型使用cuda时,会递归地让模型的参数都使用cuda。
.cuda前的,必须保证变量时Variable或者时模型。
!!!需要注意的是,如果模型用了cuda,那么如果需要用模型中的数据与外面的值做数据交换一起生成损失函数,需要将外面的那个变量也设置为cuda,要不不能交换数据。
去掉不必要的cuda占用
一般将模型,输入变量和标签使用cuda,在模型使用cuda时,会递归地让模型的参数都使用cuda。
.cuda前的,必须保证变量时Variable或者时模型。
!!!需要注意的是,如果模型用了cuda,那么如果需要用模型中的数据与外面的值做数据交换一起生成损失函数,需要将外面的那个变量也设置为cuda,要不不能交换数据。
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1068

被折叠的 条评论
为什么被折叠?