43 lines
1.6 KiB
Plaintext
43 lines
1.6 KiB
Plaintext
|
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)
|
||
|
|
CenterCrop(size=(224, 224))
|
||
|
|
ToTensor()
|
||
|
|
)
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
571
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
Epoch 0 done.
|
||
|
|
Epoch 1 done.
|
||
|
|
Epoch 2 done.
|
||
|
|
Epoch 3 done.
|
||
|
|
Epoch 4 done.
|
||
|
|
Epoch 0 done.
|
||
|
|
Traceback (most recent call last):
|
||
|
|
File "/home/ykn/cds/chapter03_Python_image_classification/main.py", line 98, in <module>
|
||
|
|
diff = c.local_train(server.global_model)
|
||
|
|
File "/home/ykn/cds/chapter03_Python_image_classification/client.py", line 62, in local_train
|
||
|
|
clipped_difference = norm_scale * (layer.data - model.state_dict()[name])
|
||
|
|
File "/home/ykn/anaconda3/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1076, in state_dict
|
||
|
|
module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
|
||
|
|
File "/home/ykn/anaconda3/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1076, in state_dict
|
||
|
|
module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
|
||
|
|
File "/home/ykn/anaconda3/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1076, in state_dict
|
||
|
|
module.state_dict(destination, prefix + name + '.', keep_vars=keep_vars)
|
||
|
|
File "/home/ykn/anaconda3/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1072, in state_dict
|
||
|
|
destination._metadata[prefix[:-1]] = local_metadata = dict(version=self._version)
|
||
|
|
KeyboardInterrupt
|