yassouali / pytorch-segmentation

:art: Semantic segmentation models, datasets and losses implemented in PyTorch.
MIT License
1.64k stars 379 forks source link

'freeze_backbone' is not defined #114

Closed IzzaRabbani closed 3 years ago

IzzaRabbani commented 3 years ago

while using deeplabv3, getting error that freeze_backbone is not defined

yassouali commented 3 years ago

just fixed this in the mast commit, simply add freeze_backbone=False in the arguments

IzzaRabbani commented 3 years ago

Screenshot from 2021-05-20 15-43-20

yassouali commented 3 years ago

you'll just need to add the input argument:

image

IzzaRabbani commented 3 years ago

Thank you