Out of memory when saving binary weights

cooooorn / Pytorch-XNOR-Net

XNOR-Net, with binary gemm and binary conv2d kernels, support both CPU and GPU.

BSD 3-Clause "New" or "Revised" License

82 stars 23 forks source link

Out of memory when saving binary weights #11

Open shashankvkt opened 4 years ago

shashankvkt commented 4 years ago

Hi, Thank you for providing this amazing code. When I ran my code on simple network, the training code crashes at binop.encode_rows(weight, bin_weight) and I get this error torch.FatalError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1518244421288/work/torch/lib/THC/generic/THCStorage.cu:58
I am using the same pytorch version to be consistent with your work. Could you please guide me on how to resolve this issue.
Thank you

shashankvkt commented 4 years ago

The issue still persists. Could you please guide me on how to solve it.