DefaultCPUAllocator: can't allocate memory: you tried to allocate 178GB

[jsoto-gladia]

when running
~/code/FullSubNet/recipes/dns_interspeech_2020$ python inference.py -C fullsubnet/inference.toml -M /home/jsoto/code/FullSubNet/recipes/dns_interspeech_2020/fullsubnet/cum_fullsubnet_best_model_218epochs.tar -O out

I get
File "/home/jsoto/anaconda3/envs/FullSubNet/lib/python3.10/site-packages/torch/nn/modules/rnn.py", line 878, in forward
result = _VF.lstm(input, hx, self._flat_weights, self.bias, self.num_layers,
RuntimeError: [enforce fail at alloc_cpu.cpp:117] err == 0. DefaultCPUAllocator: can't allocate memory: you tried to allocate 178598936936 bytes. Error code 12 (Cannot allocate memory)