Closed perrigoh closed 1 year ago
errors occur while editing: loss values become strange for vgg16, vgg13 is fine !python3 train.py --arch vgg16 --hidden_units 4096 --learning_rate 0.01 --epochs 1 Epoch: 1/1 Train Loss: -171548999680.000000, Valid Loss: -129730248704.000000, Train Accuracy: 0.009615 Epoch: 1/1 Train Loss: -171548999680.000000, Valid Loss: -148130955264.000000, Train Accuracy: 0.009615
!python3 predict.py RuntimeError: Error(s) in loading state_dict for VGG: Unexpected key(s) in state_dict: "classifier.6.weight", "classifier.6.bias".
errors occur while editing: loss values become strange for vgg16, vgg13 is fine
!python3 train.py --arch vgg16 --hidden_units 4096 --learning_rate 0.01 --epochs 1 Epoch: 1/1 Train Loss: -171548999680.000000, Valid Loss: -129730248704.000000, Train Accuracy: 0.009615 Epoch: 1/1 Train Loss: -171548999680.000000, Valid Loss: -148130955264.000000, Train Accuracy: 0.009615
!python3 predict.py RuntimeError: Error(s) in loading state_dict for VGG: Unexpected key(s) in state_dict: "classifier.6.weight", "classifier.6.bias".