search

minerva-ml / steppy-toolkit

Curated set of transformers that make your work with steppy faster and more effective :telescope:
MIT License
22 stars 9 forks source link

Cannot import name 'get_downsample_pad' #19

Open apyskir opened 5 years ago

apyskir commented 5 years ago

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