我正在使用Lua / Torch来完成深度学习任务。有时我会加载一个非常大的模型,这会导致内存成本非常高,例如:
net = torch.load('./path/to/the/model.t7')
// after this line, the GPU memory cost raises to 2800MB of 3064MB
在这种情况下,如果我想清除变量,就像在命令clear var;
的Matlab中一样?
答案 0 :(得分:4)
你可以像这样给你的变量分配一个零值:
net = nil;
对于垃圾收集,您还可以运行:
collectgarbage();
此外,使用本地前缀声明的所有变量在完成定义的代码块后都会被销毁。