Clover-Hill
/

MemoryDecoder-gpt2-small

Model card Files Files and versions

Clover-Hill commited on Sep 29

Commit

5211579

·

verified ·

1 Parent(s): 677c01c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -111,7 +111,7 @@ logger.info(f"Base Model output: {tokenizer.decode(out_ids[0], skip_special_toke
 - **Hyperparameters:**
   - Learning rate: 1e-3
   - Beta (loss balance): 0.5
-  - Training Epoch: 10
 ## Citation

 - **Hyperparameters:**
   - Learning rate: 1e-3
   - Beta (loss balance): 0.5
+  - Training Epoch: 70
 ## Citation