diff --git a/BLcourse2.3/03_one_dim_SVI.py b/BLcourse2.3/03_one_dim_SVI.py
index e03f19cc224281bf130171dc4912c009b2f1b59f..00e5d400d9e3171ab2e2c15091417289f11d1e88 100644
--- a/BLcourse2.3/03_one_dim_SVI.py
+++ b/BLcourse2.3/03_one_dim_SVI.py
@@ -138,7 +138,7 @@ class ApproxGPModel(gpytorch.models.ApproximateGP):
             self,
             Z,
             variational_distribution,
-            learn_inducing_locations=False,
+            learn_inducing_locations=True,
         )
         super().__init__(variational_strategy)
         self.mean_module = gpytorch.means.ConstantMean()
@@ -223,7 +223,9 @@ loss_func = gpytorch.mlls.VariationalELBO(
     likelihood, model, num_data=X_train.shape[0]
 )
 
-train_dl = DataLoader(TensorDataset(X_train, y_train), batch_size=len(X_train)//2, shuffle=True)
+train_dl = DataLoader(
+    TensorDataset(X_train, y_train), batch_size=128, shuffle=True
+)
 
 n_iter = 200
 history = defaultdict(list)