Closed nguyenvulong closed 10 months ago
When the frame = 1, y_pred will only have one dimension. Please see https://github.com/erprogs/GenConViT/issues/5
y_pred
This code will handle both cases when frame is 1 or larger than 1.
When the frame = 1,
y_pred
will only have one dimension. Please see https://github.com/erprogs/GenConViT/issues/5This code will handle both cases when frame is 1 or larger than 1.