Error when running torch_fedavg_mnist_lr_one_line_example.py

[Alan-JW]

Hi, @chaoyanghe, I encountered the following error when running torch_fedavg_mnist_lr_one_line_example.py, is this a version mismatch problem?

[chaoyanghe]

@Alan-JW please help to type "fedml env" and let us know your environment info.

[Alan-JW]

@chaoyanghe How should I output fedml env?

[chaoyanghe]

@Alan-JW in your fedml environement, launch a new shell window and type "fedml env". Here is the guidance: https://doc.fedml.ai/mlops/api.html

[Alan-JW]

@chaoyanghe ，This is my fedml environment！

[fedml-dimitris]

@Alan-JW Any progress on this, or you need additional help?