search

baijinglin / TS-BSmamba2

TS-BSmamba2: A TWO-STAGE BAND-SPLIT MAMBA-2 NETWORK FOR MUSIC SEPARATION
Apache License 2.0
37 stars 0 forks source link

Size of input #4

Closed ZFTurbo closed 1 month ago

ZFTurbo commented 1 month ago

Can you please provide size of input which you used?

I tried this simple script:

C = 3*44100
model = Separator().cuda()
arr = np.zeros((8, 2, C), dtype=np.float32)
x = torch.from_numpy(arr).cuda()
res = model(x)

Depending on C I have different errors: C = 3*44100

RuntimeError: shape '[912, 4, 64, 8, 64]' is invalid for input of size 120938496

C = 130560

RuntimeError: shape '[4096, 0, 64, 8, 64]' is invalid for input of size 119537664
ZFTurbo commented 1 month ago

I found the problem. I used incorrect Mamba2 version.