refactor: reorganize project structure by moving core modules and update import paths in API server
This commit is contained in:
@@ -0,0 +1,28 @@
|
||||
"""
|
||||
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}")
|
||||
Reference in New Issue
Block a user