我正在Colab上运行代码https://github.com/thiagodma/Pytorch_exs/blob/master/MultiTaskLearning/multitask_age_gender_ethnicity_resnet34.ipynb。
当我写到:
from fastai import *
from fastai.vision import *
from fastai.layers import MSELossFlat, CrossEntropyFlat
from torchvision import transforms
import warnings
warnings.filterwarnings("ignore")
files_train = get_image_files("utkface_aligned_cropped/UTKFace")
files_valid = get_image_files("utkface_aligned_cropped/crop_part1")我知道错误:
ImportError Traceback (most recent call last)
<ipython-input-4-07ceae9afcad> in <module>()
1 from fastai import *
2 from fastai.vision import *
----> 3 from fastai.layers import MSELossFlat, CrossEntropyFlat
4 from torchvision import transforms
5 import warnings
ImportError: cannot import name 'MSELossFlat' from 'fastai.layers' (/usr/local/lib/python3.7/dist-packages/fastai/layers.py)发布于 2022-07-02 04:47:09
这可能是因为您指的是使用fastai v2库的代码片段。在最新的fastai中,MSELossFlat函数可以在fastai.losses上导入。
但是,由于您正在尝试一个旧的代码片段,所以最好再现该代码的执行环境以获得预期的结果。我建议尝试在您的环境中安装fastai v2并再次执行代码。
https://stackoverflow.com/questions/72836293
复制相似问题