TypeError: __init__() got an unexpected keyword argument 'pretrained'

[fxle]

DFL-CNN <==> Part1 : prepare for parameters <==> Begin DFL-CNN <==> Part1 : prepare for parameters <==> Done DFL-CNN <==> Part2 : Load Network <==> Begin Traceback (most recent call last): File "main.py", line 180, in main() File "main.py", line 91, in main model = DFL_VGG16(k = 10, nclass = 200)
File "/home/steven/Documents/test/DFL-CNN/model/DFL.py", line 13, in init vgg16featuremap = torchvision.models.vgg16_bn(pretrained=True).features File "/usr/local/lib/python3.5/dist-packages/torchvision/models/vgg.py", line 130, in vgg16_bn return VGG(make_layers(cfg['D'], batch_norm=True), **kwargs) TypeError: init() got an unexpected keyword argument 'pretrained'

Hello dejia when I reappearance your code this error happened ,Where is the problem?

[WangJiuniu]

Maybe the version of torchvision should be updated.

[Quinniboi10]

https://forums.fast.ai/t/forward-got-an-unexpected-keyword-argument-pretrained/80582

Try this, it worked for me.

import fastai model = fastai.vision.model.YOURMODELHERE