torch_base
torch_base copied to clipboard
writer.add_image只允许传入三维图片,但logger的save_image中传入了label和pred