【问题标题】:How can I solve the problem of pytorch program?如何解决pytorch程序的问题?
【发布时间】:2020-08-20 13:52:39
【问题描述】:

我想使用 pytorch 和我自己的数据集运行 python 程序。我遇到了错误: 回溯(最近一次通话最后): 文件“C:\Users\hp\Downloads\efficient_densenet_pytorch-master\demoEmotion.py”,第 345 行,在 fire.Fire(演示) 文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\fire\core.py”,第 138 行,在 Fire component_trace = _Fire(组件,args,parsed_flag_args,上下文,名称) _Fire 中的文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\fire\core.py”,第 468 行 目标=组件。名称) _CallAndUpdateTrace 中的文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\fire\core.py”,第 672 行 组件 = fn(*varargs, **kwargs) 文件“C:\Users\hp\Downloads\efficient_densenet_pytorch-master\demoEmotion.py”,第 323 行,在演示中 n_epochs=n_epochs,batch_size=batch_size,种子=种子) 文件“C:\Users\hp\Downloads\efficient_densenet_pytorch-master\demoEmotion.py”,第 202 行,在火车中 n_epochs=n_epochs, 文件“C:\Users\hp\Downloads\efficient_densenet_pytorch-master\demoEmotion.py”,第 83 行,在 train_epoch 输出 = 模型(输入) 调用中的文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\module.py”,第 555 行 结果 = self.forward(*input, **kwargs) 文件“C:\Users\hp\Downloads\efficient_densenet_pytorch-master\models\densenet.py”,第 151 行,向前 特征 = self.features(x) 调用中的文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\module.py”,第 555 行 结果 = self.forward(*input, **kwargs) 文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\container.py”,第 100 行,向前 输入 = 模块(输入) 调用中的文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\module.py”,第 555 行 结果 = self.forward(*input, **kwargs) 文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\container.py”,第 100 行,向前 输入 = 模块(输入) 调用中的文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\module.py”,第 555 行 结果 = self.forward(*input, **kwargs) 文件“C:\Users\hp\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\pooling.py”,第 557 行,向前 self.padding、self.ceil_mode、self.count_include_pad、self.divisor_override) RuntimeError:给定输入大小:(150x1x1)。计算的输出大小:(150x0x0)。输出尺寸太小 请指导我如何解决这个问题。提前致谢!

【问题讨论】:

    标签: python tensorflow pytorch


    【解决方案1】:

    请检查图片的输入尺寸。它必须与模型中指定的完全相同。如果图像较小,您可以使用填充。

    【讨论】:

    • 非常感谢!输入大小为:torch.Size([1, 3, 3, 158])
    • 不知道你的型号就不能说什么。请记住,当您将输入传递给模型时,尺寸会变得越来越小,尤其是在使用最大池化的情况下。所以如果你输入一个小图像,输出在中间的维度非常低,使得模型的其余部分无法向下走。
    • 谢谢!请回答如何使用填充。
    • 这是一个单独的问题。提问前先搜索。例如:stackoverflow.com/questions/43391205/…
    • 非常感谢!
    猜你喜欢
    • 2019-04-14
    • 2021-01-20
    • 2023-01-03
    • 1970-01-01
    • 1970-01-01
    • 2011-12-06
    • 2020-02-26
    • 1970-01-01
    • 2014-02-15
    相关资源
    最近更新 更多