-
Notifications
You must be signed in to change notification settings - Fork 15
Open
Description
Getting following error while running ReActNet code.
File [/workspace/animesh_trailmet/experiments/binarization/ReActNet/../../../trailmet/algorithms/binarize/ReActNet.py:173](https://vscode-remote+attached-002dcontainer-002b7b22636f6e7461696e65724e616d65223a222f616e696d6573685f636f6e7461696e6572222c2273657474696e6773223a7b22686f7374223a227373683a2f2f76732d63312e63732e7569742e6e6f227d7d.vscode-resource.vscode-cdn.net/workspace/animesh_trailmet/trailmet/algorithms/binarize/ReActNet.py:173), in ReActNet.__init__(self, teacher, model, dataloaders, **kwargs)
171 self.teacher = teacher
172 self.model = model
--> 173 self.layers = model.layers_size
174 self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
175 self.dataloaders = dataloaders
File [/opt/conda/envs/py117/lib/python3.10/site-packages/torch/nn/modules/module.py:1269](https://vscode-remote+attached-002dcontainer-002b7b22636f6e7461696e65724e616d65223a222f616e696d6573685f636f6e7461696e6572222c2273657474696e6773223a7b22686f7374223a227373683a2f2f76732d63312e63732e7569742e6e6f227d7d.vscode-resource.vscode-cdn.net/opt/conda/envs/py117/lib/python3.10/site-packages/torch/nn/modules/module.py:1269), in Module.__getattr__(self, name)
1267 if name in modules:
1268 return modules[name]
-> 1269 raise AttributeError("'{}' object has no attribute '{}'".format(
1270 type(self).__name__, name))
AttributeError: 'MobileNetV2' object has no attribute 'layers_size'
- Fix model implementations in the model's folder to fix this error.
Metadata
Metadata
Assignees
Labels
No labels