Victor Phan
29 lines
819 B
Python
29 lines
819 B
Python
"""
|
|
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}")
|