Describe the bug
|
attn = torch.cat([x, mean, std], dim=1) |
If the model is trained with FP16 or BF16 mode, here will report dtype mismatch.
So, one solution is that it need add
.to(x.dtype).
Expected behaviour
None
To Reproduce
None
Environment Details
No response
Relevant Log Output
No response
Additional Context
No response
Describe the bug
speechbrain/speechbrain/lobes/models/ECAPA_TDNN.py
Line 275 in 5beaece
If the model is trained with FP16 or BF16 mode, here will report dtype mismatch.
So, one solution is that it need add
.to(x.dtype).Expected behaviour
None
To Reproduce
None
Environment Details
No response
Relevant Log Output
No response
Additional Context
No response