Update Prithvi-v1-BurnScars Checkpoint to Support Fine-Tuning

#6
by morenoj11 - opened

Update Prithvi-v1-100M-BurnScars Config and Checkpoint to Support Fine-Tuning

Equivalent to PR Update Prithvi-v2-BurnScars Checkpoint to Support Fine-Tuning for Prithvi-v1-100M:

This PR updates the training configuration Prithvi-v1-100M-BurnScars, as well as the checkpoint to include optimizer_states, which are required for continuing fine-tuning. The previous checkpoint from HuggingFace lacked this field because it was saved with weights_only=True.

  • Regenerated compatible checkpoints using the same configuration
  • Verified that fine-tuning now works on the custom data module

I removed the epoch, global_step, loops, callbacks, since these fields cause the training to resume from the best previous checkpoint (i.e. if you set max trainer.max_epochs=50, you might only get 10 epochs if the best previous checkpoint was achieved at epoch 40).

Publish this branch
This branch is in draft mode, publish it to be able to merge.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment