PyTorch Text Generation Feature Pack: Checkpointing, Beam Search, and Interactive CLI
Implements model checkpointing during training, beam search decoding for improved text generation, an interactive command-line interface for generation parameters, and a utility to count dataset tokens.
npx skills add ECNU-ICALK/AutoSkill --skill pytorch-text-generation-feature-pack-checkpointing-beam-search-a --agent claude-code
Same command for any agent — swap --agent for codex, cursor, copilot.
Weekly change comes from our own snapshots, not the repository page — it measures attention, not adoption.
# PyTorch Text Generation Feature Pack: Checkpointing, Beam Search, and Interactive CLI Implements model checkpointing during training, beam search decoding for improved text generation, an interactive command-line interface for generation parameters, and a utility to count dataset tokens. ## Prompt # Role & Objective You are a PyTorch expert specializing in NLP and text generation. Your task is to provide specific, reusable code implementations to enhance an existing PyTorch text generation training and inference pipeline. # Communication & Style Preferences - Provide clean, executable Python code snippets compatible with PyTorch. - Use standard PyTorch conventions (e.g., `model.eval()`, `torch.no_grad()`). - Ensure code is compatible with a standard PyTorch Dataset structure (e.g., accessing `dataset.pairs`, `dataset.vocab`, `dataset.idx2token`). # Operational Rules & Constraints 1. **Model Checkpointing**: - Implement logic to save the model's state dictionary (`model.state_dict()`) during the training loop. - Save the checkpoint only if the current epoch's average loss is lower than the best loss seen so far. - Save to a specified directory (e.g., 'checkpoints'), creating the d
- Prompt
- Triggers
What does the PyTorch Text Generation Feature Pack: Checkpointing, Beam Search, and Interactive CLI skill do?
Implements model checkpointing during training, beam search decoding for improved text generation, an interactive command-line interface for generation parameters, and a utility to count dataset tokens.
How do I install it?
Run `npx skills add ECNU-ICALK/AutoSkill --skill pytorch-text-generation-feature-pack-checkpointing-beam-search-a --agent claude-code` — it drops the skill into your project so the agent can pick it up. Swap the --agent value for codex, cursor or copilot if you use one of those.
Where does this skill come from?
From ECNU-ICALK/AutoSkill, a repository with 539 stars. We read it straight from the repository tree rather than a submitted listing, so what you see here is what is actually published.
Is a popular skill a good skill?
Not necessarily. Stars measure attention, not adoption — a repository can trend for a week and be abandoned. That is why we show the weekly change from our own snapshots next to the total, instead of a single flattering number.