在PopART中删除图中的损失张量

Question

我已经使用C++接口编写了一个非常简单的PopART程序，但每次我尝试编译它以在IPU设备上运行时，我都会得到以下错误：

terminate called after throwing an instance of ‘popart::error’ 
  what(): Could not find loss tensor ‘L1:0’ in main graph tensors 

我在我的程序中定义损失是这样的：

auto loss = builder->aiGraphcoreOpset1().l1loss({outputs[0]}, 0.1f, popart::ReductionType::Sum, “l1LossVal”); 

是不是我的损失定义有什么问题，导致它被从图表中删除了？我遵循了与其中一个图形核心示例here相同的结构。

Answer 1

当您传递给TrainingSession或InferenceSession对象的模型协议不包含损失张量时，通常会发生此错误。常见的原因是在将损失张量添加到图之前调用builder->getModelProto()。为了确保您的损失张量是协议的一部分，您的呼叫应按以下顺序进行：

... 
auto loss = builder->aiGraphcoreOpset1().l1loss(...); 
auto proto = builder->getModelProto(); 
auto session = popart::TrainingSession::createFromOnnxModel(...); 
... 

关键的一点是，getModelProto()调用应该是在建立PopART会话之前从builder接口进行的最后一个调用。