Kamiya

您知道TypeError: Got inappropriate size arg: (256, 256)的问题吗

1

我用了您的代码跑了您的data里的几张图片 发现会在for step, data in enumerate(data_loader['train'])这步报错，原因是 Traceback (most recent call last): File "D:/cc/segmentation_pytorch-master/main.py", line 80, in main(args) File "D:/cc/segmentation_pytorch-master/main.py", line 66, in main train(data_loader, model, optimizer, scheduler, tb_writer, param_dict, continue_epoch)...

memory leak in the training code

1

There is a memory leak in the training code of AdaPoinTr_PCN model when I train my own model, and my training memory of more than 2200epoch can occupy about 40G+.By...