22 lines
520 B
Python
22 lines
520 B
Python
#!/usr/bin/env python3
|
|
"""
|
|
Simple training runner script for AI Trainer
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
# Add src to path
|
|
sys.path.append(str(Path(__file__).parent / "src"))
|
|
|
|
from main import main
|
|
|
|
if __name__ == "__main__":
|
|
# Set environment variables for better CUDA performance
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
|
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:512'
|
|
os.environ['TOKENIZERS_PARALLELISM'] = 'false'
|
|
|
|
# Run the main training application
|
|
main() |