TypeError: alexnet() got an unexpected keyword argument 'jigsaw_classes'

[smilesun]

`python train_jigsaw.py --network alexnet

`> Traceback (most recent call last):

File "train_jigsaw.py", line 198, in
main()
File "train_jigsaw.py", line 192, in main
trainer = Trainer(args, device)
File "train_jigsaw.py", line 59, in init
model = model_factory.get_network(args.network)(jigsaw_classes=args.jigsaw_n_classes + 1, classes=args.n_classes)
File "JigenDG/models/model_factory.py", line 21, in get_network_fn
return nets_mapname
TypeError: alexnet() got an unexpected keyword argument 'jigsaw_classes'

[silvia1993]

Try to use "--network caffenet"