mega.pytorch
mega.pytorch copied to clipboard
there was a probblem using my own datasets:RuntimeError: Sizes of tensors must match except in dimension 3. Got 425 and 423 (The offending index is 0)
Does anyone konw how to solve thie problem?
Hi, did you solve this problem? I also had the same problem. This error happens only when training with RDN, FGFA, MEGA to over 20000 iterations. You should check tensor size of img['cur'] and img['ref']. I ignored all cases which get the different tensor sizes. Solved this problem.