From b4e5b15db42c334832320d34796a929f70724eec Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Oriol=20Colom=C3=A9=20Font?= <50587509+oriolcolomefont@users.noreply.github.com> Date: Sun, 13 Aug 2023 16:04:22 +0200 Subject: [PATCH] Update main.py --- main.py | 25 ++++++++++++++++++++++++- 1 file changed, 24 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index e997cf4..a24385e 100644 --- a/main.py +++ b/main.py @@ -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 = {