在某些 epoch 之后出现空张量错误
我在实现 train_domain_A.py 后收到以下类型的错误消息: > File "train_domain_A.py", line 70, in <module> losses, generated = model(Variable(data['label']), Variable(data['inst'])) File "/home/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/home/Desktop/study/Bringing-Old-Photos-Back-to-Life-master/Global/models/pix2pixHD_model_DA.py", line 196, in forward real_old_feat=torch.cat(real_old_feat,0) RuntimeError: 此函数没有接收任何张量参数(例如,您传入了一个空列表的张量),但没有为 schema aten::_cat 注册回退函数。 这通常意味着此函数需要一个非空列表的张量。 可用的函数为 [CPU, CUDA, QuantizedCPU, Autograd, Profiler, Tracer, Autocast] 我的问题是,为什么在某些 epoch 后出现了这种类型的错误。 我的假设是,这可能是由于 data['inst'] 都为 0 或 1(可能是因为我将 batchSize 设置为较低数值)而意外发生的。
内容来源: microsoft/Bringing-Old-Photos-Back-to-Life