75 lines
3.3 KiB
Plaintext
75 lines
3.3 KiB
Plaintext
/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: /home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torchvision/image.so: undefined symbol: _ZN3c104impl8GPUTrace13gpuTraceStateE
|
|
warn(f"Failed to load image Python extension: {e}")
|
|
Dataset ImageFolder
|
|
Number of datapoints: 5712
|
|
Root location: /home/ykn/cds/chapter03_Python_image_classification/data/Brain Tumor MRI Dataset/archive/Training
|
|
StandardTransform
|
|
Transform: Compose(
|
|
Resize(size=256, interpolation=bilinear, max_size=None, antialias=None)
|
|
CenterCrop(size=(224, 224))
|
|
ToTensor()
|
|
)
|
|
571
|
|
571
|
|
571
|
|
571
|
|
571
|
|
571
|
|
571
|
|
571
|
|
571
|
|
571
|
|
|
|
|
|
|
|
5
|
|
Epoch 0 done.
|
|
Epoch 1 done.
|
|
Epoch 2 done.
|
|
Epoch 0, acc: 28.756674, loss: 1.558331
|
|
|
|
2
|
|
Epoch 0 done.
|
|
Epoch 1 done.
|
|
Epoch 2 done.
|
|
Epoch 0, acc: 37.909992, loss: 1.364422
|
|
|
|
4
|
|
Epoch 0 done.
|
|
Epoch 1 done.
|
|
Epoch 2 done.
|
|
Epoch 0, acc: 46.453089, loss: 1.221337
|
|
|
|
7
|
|
Epoch 0 done.
|
|
Epoch 1 done.
|
|
Epoch 2 done.
|
|
Traceback (most recent call last):
|
|
File "cwt_main.py", line 106, in <module>
|
|
acc, loss = server.model_eval()
|
|
File "/home/ykn/cds/chapter03_Python_image_classification/server.py", line 41, in model_eval
|
|
output = self.global_model(data)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
|
|
return forward_call(*input, **kwargs)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/timm/models/vision_transformer.py", line 639, in forward
|
|
x = self.forward_features(x)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/timm/models/vision_transformer.py", line 624, in forward_features
|
|
x4 = self.blocks(x3)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
|
|
return forward_call(*input, **kwargs)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torch/nn/modules/container.py", line 141, in forward
|
|
input = module(input)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
|
|
return forward_call(*input, **kwargs)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/timm/models/vision_transformer.py", line 156, in forward
|
|
x = x + self.drop_path2(self.ls2(self.mlp(self.norm2(x))))
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
|
|
return forward_call(*input, **kwargs)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/timm/layers/mlp.py", line 43, in forward
|
|
x = self.act(x)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
|
|
return forward_call(*input, **kwargs)
|
|
File "/home/ykn/anaconda3/envs/cds/lib/python3.7/site-packages/torch/nn/modules/activation.py", line 670, in forward
|
|
return F.gelu(input)
|
|
RuntimeError: CUDA out of memory. Tried to allocate 38.00 MiB (GPU 0; 23.65 GiB total capacity; 5.56 GiB already allocated; 31.31 MiB free; 5.89 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
|