-
- Downloads
Refactor data retrieval, model I/O, and training logic for improved modularity and flexibility
- Added to support organized data downloads by robot UUID. - Enhanced and functions with typing and documentation. - Introduced functions for saving and loading configurations to/from JSON files. - Updated training workflow to handle pre-trained model layers: selectively remove last layers and freeze them before adding new layers. - Implemented flexible setup for W&B sweeps, including hyperparameter configuration from external data. - Streamlined upload process by refactoring config saving logic and reducing redundant code. - Improved logging for data analysis and sweep setup for better traceability.
Showing
- dynamics_learning/dynamics_learning/data_retrieval/__init__.py 11 additions, 0 deletions...ics_learning/dynamics_learning/data_retrieval/__init__.py
- dynamics_learning/dynamics_learning/model_io/__init__.py 59 additions, 2 deletionsdynamics_learning/dynamics_learning/model_io/__init__.py
- dynamics_learning/dynamics_learning/preprocessing/dataset_analysis.py 2 additions, 1 deletion...rning/dynamics_learning/preprocessing/dataset_analysis.py
- dynamics_learning/dynamics_learning/sweep/setup.py 84 additions, 7 deletionsdynamics_learning/dynamics_learning/sweep/setup.py
- dynamics_learning/dynamics_learning/sweep/sweep_config.py 5 additions, 1 deletiondynamics_learning/dynamics_learning/sweep/sweep_config.py
- dynamics_learning/dynamics_learning/training/__init__.py 109 additions, 16 deletionsdynamics_learning/dynamics_learning/training/__init__.py
Loading
Please register or sign in to comment