首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >无法从“MSELossFlat”导入名称“fastai.layers”

无法从“MSELossFlat”导入名称“fastai.layers”
EN

Stack Overflow用户
提问于 2022-07-02 03:41:47
回答 1查看 90关注 0票数 0

我正在Colab上运行代码https://github.com/thiagodma/Pytorch_exs/blob/master/MultiTaskLearning/multitask_age_gender_ethnicity_resnet34.ipynb

当我写到:

代码语言:javascript
复制
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")

我知道错误:

代码语言:javascript
复制
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)
EN

回答 1

Stack Overflow用户

发布于 2022-07-02 04:47:09

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

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

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/72836293

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档