"""
Script test nhanh cho cloud removal training
"""

import sys
from pathlib import Path

# Add winter_dataset to path
sys.path.insert(0, str(Path(__file__).parent / "winter_dataset"))

from train_cloud_removal import train_cloud_removal_model

if __name__ == "__main__":
    print("\n🌥️  Starting Cloud Removal Training Test")
    print("=" * 70)
    
    # Test with small dataset
    model, train_losses, val_losses = train_cloud_removal_model(
        data_dir="winter_dataset",
        use_s1=True,  # Use S1 radar data
        batch_size=4,  # Small batch for testing
        num_epochs=5,  # Few epochs for quick test
        learning_rate=1e-4
    )
    
    print("\n✅ Training test completed!")
    print(f"Final train loss: {train_losses[-1]:.6f}")
    print(f"Final val loss: {val_losses[-1]:.6f}")