【发布时间】: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