概述
我正在尝试训练从Detectron2获取的对象检测网络,并将其加载到Pytorch中。它必须检测苹果。问题是,当我使用forward()函数时,在功能中会出现以下错误:
train_one_epoch(py_model,优化器,data_loader,设备,纪元,args.print_freq)
在train_one_epoch的第30行中,文件“ /home/felipevw/MyAppleDetector/utility/engine.py”
loss_dict =模型(图像,目标)
在调用中,文件“ /home/felipevw/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py”,第532行
结果= self.forward(* input,** kwargs)
TypeError:forward()接受2个位置参数,但给出了3个
如果有人可以帮助我,我将不胜感激。我的代码是:
import datetime
import os
import time
import torch
import torch.utils.data
import torchvision
from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
from torchvision.models.detection.mask_rcnn import MaskRCNNPredictor
fr
最后
以上就是现代过客为你收集整理的python中forward的参数_forward()中的Pytorch错误需要2个位置参数,但是给出了3个...的全部内容,希望文章能够帮你解决python中forward的参数_forward()中的Pytorch错误需要2个位置参数,但是给出了3个...所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复