Your device might not have enough memory and thus you are seeing the OOM issue.
Could you explain, what noise_tunnel does and could you try to lower the batch size or number of samples you are passing to this method?