在训练和测试的时候都遇到这个问题,
_error_if_any_worker_fails()
RuntimeError: DataLoader worker (pid 8565) is killed by signal: Killed.
还有这个,
RuntimeError: $ Torch: not enough memory: you tried to allocate
经过研究,修改batch_size参数就不报错了。将batch_size改小一点,甚至是1都可以。
在训练和测试的时候都遇到这个问题,
_error_if_any_worker_fails()
RuntimeError: DataLoader worker (pid 8565) is killed by signal: Killed.
还有这个,
RuntimeError: $ Torch: not enough memory: you tried to allocate
经过研究,修改batch_size参数就不报错了。将batch_size改小一点,甚至是1都可以。