RuntimeError: CUDA out of memory. Tried to allocate … MiB

Bug记录: 方法1.改batch_size和num_worker,结果失败,考虑到计算量问题,把网络最后输出加上sigmoid 解决问题

你可能感兴趣的:(python)