feat: print model config and add evaluation notebook
This commit is contained in:
2
utils.py
2
utils.py
@@ -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):
|
||||
|
Reference in New Issue
Block a user