steppy-toolkit
steppy-toolkit copied to clipboard
Cannot import name 'get_downsample_pad'
Hi,
while trying to use UNet from scratch I get ImportError: cannot import name 'get_downsample_pad'.
This line https://github.com/neptune-ml/steppy-toolkit/blob/master/toolkit/pytorch_transformers/architectures/unet.py#L6:
from toolkit.pytorch_transformers.utils import get_downsample_pad, get_upsample_pad
should probably be changed to:
from toolkit.pytorch_transformers.architectures.utils import get_downsample_pad, get_upsample_pad