v1.0.1

What's Changed

Improvements

• Add encoder architecture args (--hidden_size, --depth, --num_attn_head, etc.) to finetune parser
• Add save_model_for_restart for finetuning training resume
• Add strict_shape_check arg to load_checkpoint
• Refactor parameter filtering to use named_parameters
• Split param_count into param_count_trainable and param_count_total

cuik-molmaker v0.2 upgrade

• Bump cuik-molmaker from 0.1.1 to 0.2

Bug fixes

• Disallow inconsistent usage between finetune and predict arguments
• Fix TypeError when dynamic_depth="truncnorm" by using rvs() instead of rvs(1)
• Fix checkpoint handling, validation, and training robustness

Testing

• Add unit tests for scheduler, nn_utils, model building, and featurization
• Ensure predict integration test uses consistent args with finetune