We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2651ee4 commit 2234981Copy full SHA for 2234981
examples/pytorch/image_recognition/torchvision_models/pruning/magnitude/eager/main.py
@@ -227,9 +227,9 @@ def train_func(model):
227
return model
228
229
prune.model = common.Model(model)
230
- prune._train_dataloader = train_loader
231
- prune._eval_dataloader = val_loader
232
- prune._pruning_func = train_func
+ prune.train_dataloader = train_loader
+ prune.eval_dataloader = val_loader
+ prune.train_func = train_func
233
model = prune.fit()
234
model.save(args.output_model)
235
return
0 commit comments