Graham Billey

Has there been any fix for this issue? I'm running into the same problem. The `monkey-patching` suggested by @mborus did not work. Thanks!

> > Has there been any fix for this issue? I'm running into the same problem. The `monkey-patching` suggested by @mborus did not work. > > Thanks! > > I...

Commenting here so I remember to check back later. Looking forward to 1.0! Thanks for making this package

Just wanted to see if any progress has been made on this yet? If there is any way I can help I'd be happy to, please let me know if...

@fkiraly The code that causes the error is this: `baseline_predictions = Baseline().predict(val_dataloader, trainer_kwargs=dict(accelerator="cpu"), return_y=True)`. Does that answer your question?

Looking into this more, there are many posts on Stack Overflow and other sources where people have the same error. Posting a few examples below as info. For example: https://stackoverflow.com/questions/77723713/runtimeerror-sizes-of-tensors-must-match-except-in-dimension-1-expected-size-1...

I think allowing a list of tags is a good idea. There should also be a tag for if the model is pre-trained, vs if the implementation is just the...