search

mx-mark / VideoTransformer-pytorch

PyTorch implementation of a collections of scalable Video Transformer Benchmarks.
272 stars 34 forks source link

AttributeError: 'VideoTransformer' object has no attribute 'weight_decay' #15

Closed Enclavet closed 2 years ago

Enclavet commented 2 years ago

I got this error until I changed the following model_trainer.py:

param_group["weight_decay"] = self._get_momentum(base_value=self.weight_decay, final_value=self.configs.weight_decay_end)

to

param_group["weight_decay"] = self._get_momentum(base_value=self.configs.weight_decay, final_value=self.configs.weight_decay_end)

mx-mark commented 2 years ago

@Enclavet Thanks for your reporting about this problem.