diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py index efe1254..71b4574 100644 --- a/pytorch/pytorch.py +++ b/pytorch/pytorch.py @@ -140,7 +140,6 @@ output_size = len(output_cols) def my_config(): global epochs epochs = numberOfEpochParam - lr=lr model=model train_loader=train_loader val_loader=val_loader