I’m training a classification model based on Resnet34 to classify satellite images based on one of three classes. Loading the data and setting up the model works fine, I can also train the model successfully. However, the metrics are always constant (tried only accuracy and both accuracy and error_rate). I use these commands:
model = vision_learner(dls = data, arch = models.resnet34, metrics = accuracy)
model.fine_tune(epochs = 1)
In addition, fine_tune doesn’t stop after 1 epoch. Could or is this related to the issue with the metrics?
I’m using fastAI 2.7.12 and PyTorch 2.0.1 on an Apple M1 and the code is running in Jupyter Notebook.
Thanks!