flpelerin commited on
Commit
5f83172
·
1 Parent(s): d81727e

Update file model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -3
model.py CHANGED
@@ -18,15 +18,15 @@ class Model:
18
  print(f"Model's embedding size is {self.params.vocab_size}")
19
 
20
 
21
- def parameters():
22
  return self.model.parameters()
23
 
24
 
25
- def unfreeze():
26
  self.model.train()
27
 
28
 
29
- def freeze():
30
  self.model.eval()
31
 
32
 
 
18
  print(f"Model's embedding size is {self.params.vocab_size}")
19
 
20
 
21
+ def parameters(self):
22
  return self.model.parameters()
23
 
24
 
25
+ def unfreeze(self):
26
  self.model.train()
27
 
28
 
29
+ def freeze(self):
30
  self.model.eval()
31
 
32