diff --git a/opengait/modeling/models/gaitgl.py b/opengait/modeling/models/gaitgl.py index 19953f5..bd4d7ce 100644 --- a/opengait/modeling/models/gaitgl.py +++ b/opengait/modeling/models/gaitgl.py @@ -193,7 +193,7 @@ class GaitGL(BaseModel): 'image/sils': sils.view(n*s, 1, h, w) }, 'inference_feat': { - 'embeddings': bnft + 'embeddings': embed } } return retval