Skip to content
This repository has been archived by the owner on Jul 22, 2024. It is now read-only.

UnboundLocalError: local variable 'shape_estimator' referenced before assignment #2

Open
zarekxu opened this issue May 27, 2020 · 2 comments

Comments

@zarekxu
Copy link

zarekxu commented May 27, 2020

Thanks for sharing your great work. When I reimplemented your code, I met the following error:
UnboundLocalError: local variable 'shape_estimator' referenced before assignment

I used LeNet on Mnist.

Could you tell me the reason may cause such error?

Thanks.

@hwang595
Copy link
Contributor

Hi @zarekxu, thanks a lot for your interest in our work! The shape_estimator is used to calculate the input dimension of the very first FC layer in a CNN model architecture e.g. LeNet or VGG. Based on your description, it seems there are some issues with model weight reloading in the MNIST+LeNet experiment.

Can you please provide more details on how to reproduce the issue e.g. what was launch script you used and which exact line reports that error?

Thanks again and I'm happy to provide more help on this issue.

@makhanov-nu
Copy link

Hello Authors,
I ran the vgg on Mnist and got the "shape_estimator error. I didn't change anything in the code, just wanted to recreate the results of the paper. Here is the content of the run.sh file:

python main.py --model=vgg
--dataset=cifar10
--lr=0.01
--retrain_lr=0.01
--batch-size=64
--epochs=20
--retrain_epochs=20
--n_nets=16
--partition=hetero-dir
--comm_type=fedma
--comm_round=10
--oneshot_matching=
--retrain=True

So, can you tell me what is the issue?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants