torch:1.10.0+cu111
pandas:1.3.3
报错:
定位:
df = pd.DataFrame(columns=[epoch, model.sigma1.cpu().detach().numpy(), model.sigma2.cpu().detach().numpy(),model.sigma3.cpu().detach().numpy(),iter,avg_ssim, avg_psnr,model.current_log()])
解决方法:pandas升级至1.4.4以上
detach是把torch计算过程中数据里的梯度取消,这样可以用dataframe方法写入csv,如果不用detach会提示,但是用了就会终止,生成core.xx,Segmentation fault 问题经常是内存溢出,但实际查看内存没有超过,就升级了pandas,然后问题解决。