CognitiveMemoryLayer-models / fact_extraction_structured_epoch_stats.json
avinashm's picture
Upload all trained models
e65379c verified
{
"task": "fact_extraction_structured",
"epoch_stats": [
{
"epoch": 1,
"train_loss": 0.1299236143381422
},
{
"epoch": 2,
"train_loss": 0.0057935971935330595
},
{
"epoch": 3,
"train_loss": 0.0012633003406408534
}
],
"training_summary": {
"actual_epochs": 3,
"best_epoch": 3,
"early_stopped": false
},
"labels": {
"0": "O",
"1": "B-attribute",
"2": "I-attribute",
"3": "B-causal",
"4": "I-causal",
"5": "B-goal",
"6": "I-goal",
"7": "B-identity",
"8": "I-identity",
"9": "B-location",
"10": "I-location",
"11": "B-occupation",
"12": "I-occupation",
"13": "B-policy",
"14": "I-policy",
"15": "B-preference",
"16": "I-preference",
"17": "B-state",
"18": "I-state",
"19": "B-value",
"20": "I-value"
},
"hf_model_dir": "/nvme/CognitiveMemoryLayer/packages/models/trained_models/fact_extraction_structured_hf"
}