Update train_utils.py

train_utils.py

@@ -127,7 +127,7 @@ def collect_predictions(model, loader, device):
 def train_model(
     dropout: float = 0.4,
     fc_dim: int = 256,
-    learning_rate: float = 0.00001,
+    learning_rate: float = 0.00005,
     weight_decay: float = 0.0001,
     batch_size: int = 16,
     epochs: int = 30,