Как решить проблему с программой pytorch? - PullRequest
0 голосов
/ 05 мая 2020

Я хочу запустить программу python с использованием pytorch с моим собственным набором данных. Я столкнулся с ошибкой: Traceback (последний вызов последний): File "C: \ Users \ hp \ Downloads \ effective_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 (component, args, parsed_flag_args, context, name) File "C: \ Users \ hp \ Anaconda3 \ envs \ tf-gpu \ lib \ site-packages \ fire \ core.py", строка 468 , в _Fire target = component. name ) File "C: \ Users \ hp \ Anaconda3 \ envs \ tf-gpu \ lib \ site-packages \ fire \ core.py", строка 672, в _CallAndUpdateTrace component = fn (* varargs, ** kwargs) File "C: \ Users \ hp \ Downloads \ effective_densenet_pytorch-master \ demoEmotion.py", строка 323, в демонстрации n_epochs = n_epochs, seed = batch_size = batch_size seed) Файл "C: \ Users \ hp \ Downloads \ effective_densenet_pytorch-master \ demoEmotion.py", строка 202, в поезде n_epochs = n_epochs, File "C: \ Users \ hp \ Downl oads \ effective_densenet_pytorch-master \ demoEmotion.py ", строка 83, в train_epoch output = model (input) File" C: \ Users \ hp \ Anaconda3 \ envs \ tf-gpu \ lib \ site-packages \ torch \ nn \ modules \ module.py ", строка 555, в call result = self.forward (* input, ** kwargs) File" C: \ Users \ hp \ Downloads \ effective_densenet_pytorch-master \ models \ dens enet .py ", строка 151, in forward features = self.features (x) File" C: \ Users \ hp \ Anaconda3 \ envs \ tf-gpu \ lib \ site-packages \ torch \ nn \ modules \ module.py ", строка 555, в call result = self.forward (* input, ** kwargs) File" C: \ Users \ hp \ Anaconda3 \ envs \ tf-gpu \ lib \ site-packages \ torch \ nn \ modules \ container.py ", строка 100, в прямом вводе = module (input) File" C: \ Users \ hp \ Anaconda3 \ envs \ tf-gpu \ lib \ site-packages \ torch \ nn \ modules \ module.py ", строка 555, в call result = self.forward (* input, ** kwargs) File" C: \ Users \ hp \ Anaconda3 \ envs \ tf-gpu \ lib \ site-packages \ torch \ nn \ modules \ container.py ", строка 100, in forward input = module (inpu t) Файл «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 , in forward self.padding, self.ceil_mode, self.count_include_pad, self.divisor_override) RuntimeError: данный размер ввода: (150x1x1). Расчетный выходной размер: (150x0x0). Размер вывода слишком мал. Подскажите, как решить эту проблему. Заранее спасибо!

1 Ответ

0 голосов
/ 06 мая 2020

Пожалуйста, проверьте размер изображения. Он должен быть точно таким, как указано в модели. Вы можете использовать отступы, если изображение меньше.

...