Pausing for now, train loop should now work and adding some tests
This commit is contained in:
25
configs/debug_config.py
Normal file
25
configs/debug_config.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from configs.base_config import base_config
|
||||
|
||||
# Create a debug configuration with minimal settings
|
||||
config = base_config.copy()
|
||||
|
||||
# Update settings for quick debugging
|
||||
config.update(
|
||||
{
|
||||
# Core configuration
|
||||
"config_name": "debug_run",
|
||||
"data_root": "data/PennFudanPed",
|
||||
"num_classes": 2, # background + pedestrian
|
||||
# Minimal training parameters
|
||||
"batch_size": 1,
|
||||
"num_epochs": 1, # Just one epoch for testing
|
||||
"val_split_ratio": 0.2, # Use more validation samples for better testing coverage
|
||||
# Performance optimizations
|
||||
"pin_memory": False,
|
||||
"num_workers": 0, # Use 0 workers to avoid multiprocessing complexities during debugging
|
||||
# Logging settings
|
||||
"log_freq": 1, # Log every batch for debugging
|
||||
# Device setting - use CPU for reliable debugging
|
||||
"device": "cpu", # Using CPU ensures consistent behavior across systems
|
||||
}
|
||||
)
|
||||
Reference in New Issue
Block a user