I use this command to run the code python main.py --config_env configs/env.yml --config_exp configs/$DATASET/$MODEL.yml
since I can only install fastmoe==0.1.2. However, when I run with the command above. the following errors happened.
Traceback (most recent call last):
File "main.py", line 13, in
from utils.common_config import get_train_dataset, get_transformations,
File "/home/shawn/local/szz/workspace/M3ViT/utils/common_config.py", line 24, in
from utils.moe_utils import read_specific_group_experts
File "/home/shawn/local/szz/workspace/M3ViT/utils/moe_utils.py", line 2, in
from models.gate_funs.noisy_gate import NoisyGate
File "/home/shawn/local/szz/workspace/M3ViT/models/gate_funs/noisy_gate.py", line 4, in
from fmoe.gates.base_gate import BaseGate
File "/home/shawn/anaconda3/envs/x/lib/python3.8/site-packages/fastmoe-0.1.2-py3.8-linux-x86_64.egg/fmoe/init.py", line 5, in
from .layers import FMoELinear, FMoE
File "/home/shawn/anaconda3/envs/x/lib/python3.8/site-packages/fastmoe-0.1.2-py3.8-linux-x86_64.egg/fmoe/layers.py", line 10, in
from .gates import NaiveGate
File "/home/shawn/anaconda3/envs/x/lib/python3.8/site-packages/fastmoe-0.1.2-py3.8-linux-x86_64.egg/fmoe/gates/init.py", line 8, in
from .gshard_gate import GShardGate
File "/home/shawn/anaconda3/envs/x/lib/python3.8/site-packages/fastmoe-0.1.2-py3.8-linux-x86_64.egg/fmoe/gates/gshard_gate.py", line 8, in
from .utils import limit_by_capacity
File "/home/shawn/anaconda3/envs/x/lib/python3.8/site-packages/fastmoe-0.1.2-py3.8-linux-x86_64.egg/fmoe/gates/utils.py", line 5, in
from fmoe.functions import count_by_gate
ImportError: cannot import name 'count_by_gate' from 'fmoe.functions' (/home/shawn/anaconda3/envs/x/lib/python3.8/site-packages/fastmoe-0.1.2-py3.8-linux-x86_64.egg/fmoe/functions.py)
VITA-Group
