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:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user