docs: update train method docstring

crux-bphc · Aug 30, 2023 · a41483c · a41483c
1 parent 03c0161
commit a41483c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/clusterer/model.py b/clusterer/model.py
@@ -112,6 +112,7 @@ def train(self, train_data: TripletDataset, batch_size : int, n_epochs : int, le
             batch_size (int)
             n_epochs (int)
             learning_rate (float)
+            frozen (int) : Number of layers upto which parameters are to be froze. defaults to 250
 
         Returns:
             None