This repository was archived by the owner on Jun 22, 2022. It is now read-only.
This repository was archived by the owner on Jun 22, 2022. It is now read-only.
UNet error #2
Description
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.