Problem whe trying the lines on trying the Demo Examples Usage: IndexError: "Dimension out of range (expected to be in range of [-2, 1], but got 2)"

[AlefPontesSilva]

Hello!
My name is Alef de Pontes Silva. I'm currently studying Control and Automation Engineering in the institution CEFET/RJ - Nova Iguaçu (RJ, Brazil). I'm almost finishing college, and my final project relates to the ideia of the project of this repository.

So, I got all the material you let in github (https://github.com/gulvarol/bsldict) for learning it and aply it in our case (in the brazilian sign language, Libras) and tried to run the exemples on demo:

but I always get the same error:

Have someone ever see this? As all the material is on the computer, I think it should work. If you could help me, I'd be very grateful!

[mochaminte]

I'm also getting this issue

[mochaminte]

I found a fix for this issue, replace:

continuous_features = np.append(
    continuous_features,
    out["embds"].cpu().detach().squeeze(2).squeeze(2).squeeze(2).numpy(),
    axis=0
)

with

continuous_features = np.append(
            continuous_features, out["embds"].cpu().detach().squeeze(-1).numpy(), axis=0
)