code style fixes

LostOxygen · Jun 16, 2023 · 39d7c86 · 39d7c86
1 parent 34b51cb
commit 39d7c86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kernel_eval/train.py b/kernel_eval/train.py
@@ -85,7 +85,7 @@ def train_model(model: nn.Module, train_dataloader: IterableDataset,
     model = model.to(device)
     loss_fn = nn.BCELoss()
     optimizer = optim.Adam(model.parameters(), lr=learning_rate, weight_decay=5e-4)
-    scheduler = ReduceLROnPlateau(optimizer, mode='min', factor=0.9, patience=5, verbose=True)
+    scheduler = ReduceLROnPlateau(optimizer, mode="min", factor=0.9, patience=5, verbose=True)
 
 
     for epoch in range(0, epochs):