ScalableBDL icon indicating copy to clipboard operation
ScalableBDL copied to clipboard

Update finetune_cifar.py

Open lan-qing opened this issue 3 years ago • 0 comments

Hello, thudzj,

In Line 17, ScalableBDL/reproduction/finetune_cifar.py: from scalablebdl.converter import to_bayesian, to_deterministic

However, I found the two functions, to_bayesian and to_deterministic are defined in scalablebdl.mean_field.converter.py and declared in scalablebdl.mean_field.__init__.py. Therefore, following the codes in ScalableBDL/reproduction/finetune_face.py and ScalableBDL/reproduction/finetune_imagenet.py I modified it as follows.

from scalablebdl.mean_field import PsiSGD, to_bayesian, to_deterministic.

Best

Lan-qing

lan-qing avatar Aug 02 '21 05:08 lan-qing