litian96 / FedProx

Federated Optimization in Heterogeneous Networks (MLSys '20)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ModuleNotFoundError: No module named 'FedML'

mustain-billah opened this issue · comments

After running this: !python experiments/centralized/moleculenet/molecule_classification_multilabel.py

Getting this Error Message:
Traceback (most recent call last):
File "experiments/centralized/moleculenet/molecule_classification_multilabel.py", line 11, in
from data_preprocessing.molecule.data_loader import get_dataloader, get_data
File "/content/drive/My Drive/Colab Notebooks/FedGraphNN/data_preprocessing/molecule/data_loader.py", line 12, in
from FedML.fedml_core.non_iid_partition.noniid_partition import partition_class_samples_with_dirichlet_distribution
ModuleNotFoundError: No module named 'FedML'

From the error msg, I assume you are referring to some other repo (rather than FedProx)?