Improve Glow Model Training Script

[AdnanKahveci]

Implemented a complete training pipeline for the Glow model, including:

• Data Loading: Added dynamic data loading with image transformations.
  • Training Loop: Includes loss calculation, learning rate scheduling, and early stopping.
  • Validation: Periodically validates the model and saves checkpoints.
  • Model Checkpointing: Saves model and optimizer state at intervals.
• Early Stopping: Halts training when no improvement is observed over a specified number of iterations.

  Added argparse for easier configuration of:

  • Batch size
  • Maximum iterations
  • Number of samples
  • Image directory path

[AdnanKahveci]

Refactored the Glow model training script. Improved the training loop, implemented early stopping, validation, and checkpoint saving. Added argparse for better flexibility and model configuration. The script now handles both CPU and GPU seamlessly