오류 로그
Traceback (most recent call last):
File "train.py", line 140, in <module>
output = model(X_batch)
File "/opt/conda/envs/medT/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/data/5.MedT/Medical-Transformer/lib/models/axialnet.py", line 507, in forward
return self._forward_impl(x)
File "/data/5.MedT/Medical-Transformer/lib/models/axialnet.py", line 476, in _forward_impl
x = self.bn1(x)
File "/opt/conda/envs/medT/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/opt/conda/envs/medT/lib/python3.6/site-packages/torch/nn/modules/batchnorm.py", line 148, in forward
self.num_batches_tracked = self.num_batches_tracked + 1 # type: ignore[has-type]
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
