无法导入名称＆＃x27; mselossflat＆＃x27;从fastai.layers＆＃x27;

Question

我正在运行代码 https://github.com/thiagodma/pytorch_exs/blob/master/master/multitasklearning/multitask_age_gender_ethnicity_ethnicity_resnet34.ipynb colab上。

当我写的时候：

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)

Answer 1

这可能是因为您指的是使用fastai v2库的代码段。在最新的fastai中，mselossflat函数可以在fastai.losses上导入。

但是，由于您正在尝试一个旧的代码段，因此最好重现该代码的执行环境以获得预期的结果。我建议尝试在您的环境上安装fastai v2并再次执行代码。