ScalableBDL
ScalableBDL copied to clipboard
Update finetune_cifar.py
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