GaitGL model inference issue solved

inference feature is wrongly used, causing zero accuracies. Solved by replacing to right embedding feature.
This commit is contained in:
noahshen98
2022-09-14 10:02:01 +08:00
committed by GitHub
parent cd7ccb69e7
commit 54b4d697e6
+1 -1
View File
@@ -193,7 +193,7 @@ class GaitGL(BaseModel):
'image/sils': sils.view(n*s, 1, h, w) 'image/sils': sils.view(n*s, 1, h, w)
}, },
'inference_feat': { 'inference_feat': {
'embeddings': bnft 'embeddings': embed
} }
} }
return retval return retval