feat: print model config and add evaluation notebook

This commit is contained in:
2025-10-20 10:14:50 +08:00
parent 6b782b86e1
commit 1c9e2a2fb3
2 changed files with 343 additions and 0 deletions

View File

@@ -122,6 +122,8 @@ def load_model(config_path, model_path, vocab_size, device='cpu'):
with open(config_path, 'r') as f:
config_dict = json.load(f)
print(f"Model config: {config_dict}")
# Create a config object from the dictionary
class AttrDict(dict):
def __init__(self, *args, **kwargs):