[Update variants with parameter count equation]
This commit is contained in:
parent
f00d10b59f
commit
a825ba217f
4 changed files with 72 additions and 22 deletions
10
variants_example.py
Normal file
10
variants_example.py
Normal file
|
|
@ -0,0 +1,10 @@
|
|||
from open_mythos import (
|
||||
mythos_1b,
|
||||
OpenMythos,
|
||||
)
|
||||
|
||||
cfg = mythos_1b()
|
||||
model = OpenMythos(cfg)
|
||||
|
||||
total = sum(p.numel() for p in model.parameters())
|
||||
print(f"Parameters: {total:,}")
|
||||
Loading…
Add table
Add a link
Reference in a new issue