SmolFactory / trackio.py
Tonic's picture
adds monkey patch for trackio monitoring in torch and readme creator improvements
39db0ca verified
raw
history blame
631 Bytes
"""
Trackio Module for TRL Library Compatibility
This module provides the interface expected by TRL library while using our custom monitoring system
"""
# Import all functions from our custom trackio implementation
from src.trackio import (
init,
log,
finish,
log_config,
log_checkpoint,
log_evaluation_results,
get_experiment_url,
is_available,
get_monitor
)
# Make all functions available at module level
__all__ = [
'init',
'log',
'finish',
'log_config',
'log_checkpoint',
'log_evaluation_results',
'get_experiment_url',
'is_available',
'get_monitor'
]