This repository has been archived by the owner on Jun 22, 2022. It is now read-only.

UNet error #2

Open

kamil-kaczmarek opened this issue May 23, 2018 · 0 comments

Labels

bug

Member

kamil-kaczmarek commented May 23, 2018 •

edited

Loading

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.

kamil-kaczmarek added the bug label

bcottman mentioned this issue

Feature selection by importance #30

Open

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.