Currently UNet model in its forward() method returns a single tensor, so len(tensor) is batch size. On the other hand we have assertion in: https://github.com/minerva-ml/steppy-toolkit/blob/master/steppy_toolkit/pytorch/models.py#L94, so when batch_size>1 this assertion throws an error and it should not.
Currently UNet model in its forward() method returns a single tensor, so len(tensor) is batch size. On the other hand we have assertion in:
https://github.com/minerva-ml/steppy-toolkit/blob/master/steppy_toolkit/pytorch/models.py#L94, so when batch_size>1 this assertion throws an error and it should not.