pytorch中的小bug

只用一個卡可用時，要在後邊添加逗號 ， parser.add_argument('-d', '--sys_device_ids', type=eval, default=(0,)) 修改爲 ValueError: Expected more than 1 value per channel when training, got input size [1, 256, 1, 1] 可能是輸入批次只有

>>阅读原文<<