Skip to content

Commit

Permalink
Update main.py
Browse files Browse the repository at this point in the history
  • Loading branch information
oriolcolomefont authored Aug 13, 2023
1 parent 5eceb09 commit b4e5b15
Showing 1 changed file with 24 additions and 1 deletion.
25 changes: 24 additions & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,31 @@
import eval
from tqdm import tqdm

logging.basicConfig(level=logging.INFO)
"""
Automated Experimentation and Evaluation Script
This script automates the process of training and evaluating a model under various settings.
It conducts experiments by testing different combinations of hyperparameters, training the model,
and evaluating its performance using the SALAMI dataset for boundary detection.
Usage:
python script_name.py
Dependencies:
- PyTorch
- PyTorch Lightning
- NumPy
- Pandas
- tqdm
Module Dependencies:
- train (Contains model training functionality)
- features (Contains feature extraction functionality)
- eval (Contains evaluation functionality)
"""

logging.basicConfig(level=logging.INFO)

def create_settings():
settings_dict = {
Expand Down

0 comments on commit b4e5b15

Please sign in to comment.