transformers safetensors codebook generated_from_trainer dataset:roneneldan/TinyStories base_model:roneneldan/TinyStories-1Layer-21M model-index endpoints_compatible region:us
---
base_model: roneneldan/TinyStories-1Layer-21M
tags:
- generated_from_trainer
datasets:
- roneneldan/TinyStories
metrics:
- accuracy
model-index:
- name: tinystories_1layer_attn_mlp_C10k_k100
  results:
  - task:
      name: Causal Language Modeling
      type: text-generation
    dataset:
      name: roneneldan/TinyStories
      type: roneneldan/TinyStories
    metrics:
    - name: Accuracy
      type: accuracy
      value: 0.5429091526514649
---

<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->

# tinystories_1layer_attn_mlp_C10k_k100

This model is a fine-tuned version of [roneneldan/TinyStories-1Layer-21M](https://huggingface.co/roneneldan/TinyStories-1Layer-21M) on the roneneldan/TinyStories dataset.
It achieves the following results on the evaluation set:
- Loss: 1.8957
- Accuracy: 0.5429
- Multicode K: 1
- Dead Code Fraction/layer0: 0.0
- Mse/layer0: 611.1572
- Input Norm/layer0: 31.9975
- Output Norm/layer0: 15.0872

## Model description

More information needed

## Intended uses & limitations

More information needed

## Training and evaluation data

More information needed

## Training procedure

### Training hyperparameters

The following hyperparameters were used during training:
- learning_rate: 0.0005
- train_batch_size: 8
- eval_batch_size: 8
- seed: 42
- gradient_accumulation_steps: 6
- total_train_batch_size: 48
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- lr_scheduler_warmup_ratio: 0.05
- training_steps: 10000

### Training results

| Training Loss | Epoch | Step  | Validation Loss | Accuracy | Multicode K | Dead Code Fraction/layer0 | Mse/layer0 | Input Norm/layer0 | Output Norm/layer0 |
|:-------------:|:-----:|:-----:|:---------------:|:--------:|:-----------:|:-------------------------:|:----------:|:-----------------:|:------------------:|
| 2.5072        | 0.05  | 500   | 2.4764          | 0.4579   | 1           | 0.0                       | 841.1602   | 31.9977           | 4.9114             |
| 2.2285        | 0.1   | 1000  | 2.2265          | 0.4926   | 1           | 0.0                       | 792.3023   | 31.9980           | 7.5524             |
| 2.1472        | 0.16  | 1500  | 2.1584          | 0.5025   | 1           | 0.0                       | 761.8683   | 31.9980           | 8.9239             |
| 2.1144        | 0.21  | 2000  | 2.1128          | 0.5090   | 1           | 0.0                       | 737.1843   | 31.9979           | 9.8992             |
| 2.0847        | 0.26  | 2500  | 2.0791          | 0.5142   | 1           | 0.0                       | 716.9390   | 31.9979           | 10.6577            |
| 2.0439        | 0.31  | 3000  | 2.0482          | 0.5185   | 1           | 0.0                       | 698.7266   | 31.9979           | 11.3599            |
| 2.0263        | 0.37  | 3500  | 2.0253          | 0.5224   | 1           | 0.0                       | 682.2680   | 31.9979           | 12.0105            |
| 1.9906        | 0.42  | 4000  | 2.0066          | 0.5253   | 1           | 0.0                       | 669.1965   | 31.9979           | 12.5568            |
| 1.9852        | 0.47  | 4500  | 1.9898          | 0.5279   | 1           | 0.0                       | 657.5872   | 31.9979           | 13.0526            |
| 1.9687        | 0.52  | 5000  | 1.9757          | 0.5300   | 1           | 0.0                       | 648.2462   | 31.9979           | 13.4496            |
| 1.9672        | 0.57  | 5500  | 1.9620          | 0.5321   | 1           | 0.0                       | 640.0822   | 31.9978           | 13.8078            |
| 1.9441        | 0.63  | 6000  | 1.9513          | 0.5339   | 1           | 0.0                       | 633.8831   | 31.9978           | 14.1018            |
| 1.9408        | 0.68  | 6500  | 1.9397          | 0.5358   | 1           | 0.0                       | 628.0929   | 31.9977           | 14.3550            |
| 1.9256        | 0.73  | 7000  | 1.9302          | 0.5374   | 1           | 0.0                       | 623.2726   | 31.9977           | 14.5534            |
| 1.9204        | 0.78  | 7500  | 1.9225          | 0.5381   | 1           | 0.0                       | 619.4573   | 31.9977           | 14.7258            |
| 1.907         | 0.84  | 8000  | 1.9150          | 0.5393   | 1           | 0.0                       | 616.4379   | 31.9976           | 14.8625            |
| 1.8931        | 0.89  | 8500  | 1.9076          | 0.5408   | 1           | 0.0                       | 613.7874   | 31.9976           | 14.9685            |
| 1.9021        | 0.94  | 9000  | 1.9021          | 0.5417   | 1           | 0.0                       | 612.0126   | 31.9975           | 15.0379            |
| 1.8967        | 0.99  | 9500  | 1.8970          | 0.5426   | 1           | 0.0                       | 610.6121   | 31.9975           | 15.0932            |
| 1.8942        | 1.04  | 10000 | 1.8957          | 0.5429   | 1           | 0.0                       | 611.1572   | 31.9975           | 15.0872            |


### Framework versions

- Transformers 4.35.2
- Pytorch 2.1.0+cu121
- Datasets 2.16.1
- Tokenizers 0.15.1
