added function count_parameters

Author: n0obcoder

utils_modified.py

@@ -4,6 +4,11 @@
 def q():
     sys.exit()
 
+# define a function to count the total number of trainable parameters
+def count_parameters(model): 
+    num_parameters = sum(p.numel() for p in model.parameters() if p.requires_grad)
+    return num_parameters/1e6 # in terms of millions
+
 # TEST
 def nearest_word(inp, emb, top = 5, debug = False):
     euclidean_dis = np.linalg.norm(inp - emb, axis = 1)    
@@ -18,4 +23,4 @@ def nearest_word(inp, emb, top = 5, debug = False):
         print('emb_ranking: ', emb_ranking)
         print(f'top {top} embeddings are: {emb_ranking[:top]} with respective distances\n {euclidean_dis_top}')
 
-    return emb_ranking_top, euclidean_dis_top
+    return emb_ranking_top, euclidean_dis_top