Closed bruAristimunha closed 4 months ago
@sliwy not sure all the models can be initialized with their default parameters.
Plus I think we should test if the models accept inputs with the desired shape and return outputs with the correct shape:
from braindecode.models.util import models_dict @pytest.mark.parametrize("model_cls", models_dict.values()) def test_model_torchinfo(model_cls): model = model_cls(n_times=5000, n_chans=1, n_outputs=2) assert model(torch.ones(3, 1, 5000)).shape == (3, 2)
Originally posted by @PierreGtch in https://github.com/braindecode/braindecode/issues/538#issuecomment-1725468288
Plus I think we should test if the models accept inputs with the desired shape and return outputs with the correct shape:
Originally posted by @PierreGtch in https://github.com/braindecode/braindecode/issues/538#issuecomment-1725468288