diff --git a/codebase/main.py b/codebase/main.py index a0c693b..ae7a217 100644 --- a/codebase/main.py +++ b/codebase/main.py @@ -60,7 +60,7 @@ def excute_pipeline( metric_store += evaluate_one_epoch( epoch=0, model=model, - loader=train_loader, + loader=val_loader, criterion=criterion, optimizer=optimizer, scheduler=scheduler,