Skip to content
Snippets Groups Projects
Select Git revision
1 result Searching

benchmark_number_of_runs.py

  • Leon Michel Gorißen's avatar
    aba59654
    Refactor benchmark script to modularize training workflow for multiple models · aba59654
    Leon Michel Gorißen authored
    - Added functionality to prepare data and execute training loops for various models (LLT, ITA).
    - Implemented data analysis and interpolation for trajectory data.
    - Integrated W&B sweeps to dynamically adjust training based on validation loss thresholds.
    - Enabled saving models upon reaching the loss threshold.
    - Improved logging and modular handling of multiple model configurations and UUIDs.
    - Removed old hardcoded training loops and replaced with flexible, scalable partial functions for different models.
    aba59654
    History
    Refactor benchmark script to modularize training workflow for multiple models
    Leon Michel Gorißen authored
    - Added functionality to prepare data and execute training loops for various models (LLT, ITA).
    - Implemented data analysis and interpolation for trajectory data.
    - Integrated W&B sweeps to dynamically adjust training based on validation loss thresholds.
    - Enabled saving models upon reaching the loss threshold.
    - Improved logging and modular handling of multiple model configurations and UUIDs.
    - Removed old hardcoded training loops and replaced with flexible, scalable partial functions for different models.