While the calc scripts are correct for llama-style models, their implementation is inflexible (see https://github.com/EleutherAI/cookbook/issues/36 and https://github.com/EleutherAI/cookbook/pull/35) It'd be nice to clean this up a bit.
