├── README.md ├── configs ├── config_linear.json └── config_linear_prod.json ├── models ├── __init__.py └── linear_model.py ├── torch_datasets └── product.py └── training └── linear_decoder.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/emalach/LinearLM/HEAD/README.md -------------------------------------------------------------------------------- /configs/config_linear.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/emalach/LinearLM/HEAD/configs/config_linear.json -------------------------------------------------------------------------------- /configs/config_linear_prod.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/emalach/LinearLM/HEAD/configs/config_linear_prod.json -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/linear_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/emalach/LinearLM/HEAD/models/linear_model.py -------------------------------------------------------------------------------- /torch_datasets/product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/emalach/LinearLM/HEAD/torch_datasets/product.py -------------------------------------------------------------------------------- /training/linear_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/emalach/LinearLM/HEAD/training/linear_decoder.py --------------------------------------------------------------------------------