AgentStack
SKILL verified MIT Self-run

Pytorch Training

skill-thada2402-autoresearchclaw-pytorch-training · by thada2402

Best practices for building robust PyTorch training loops. Use when generating or reviewing ML training code.

No reviews yet
0 installs
4 views
0.0% view→install

Install

$ agentstack add skill-thada2402-autoresearchclaw-pytorch-training

✓ scanned · ✓ verified — works with Claude Code, Cursor, and more.

Security review

✓ Passed

No issues found. Passed automated security review. · v0.1.0 How review works →

  • Prompt-injection patterns
  • Secret / credential exfiltration
  • Dangerous shell & filesystem operations
  • Untrusted network calls
  • Known-malicious package signatures

What it can access

  • Network access No
  • Filesystem access No
  • Shell / process execution No
  • Environment & secrets No
  • Dynamic code execution No

From automated source analysis of v0.1.0. “Used” means the capability is present in the source — more access means more to trust, not that it’s unsafe.

Are you the author of Pytorch Training? Claim this listing to set pricing, connect Stripe payouts, and keep 70% of every sale.
Sign up to claim

About

PyTorch Training Best Practice

  1. Use torch.manual_seed() for reproducibility (set for torch, numpy, random)
  2. Use DataLoader with numworkers>0 and pinmemory=True for GPU
  3. Enable cudnn.benchmark=True for fixed input sizes
  4. Use learning rate schedulers (CosineAnnealingLR or OneCycleLR)
  5. Implement early stopping based on validation metric
  6. Log metrics every epoch, save best model checkpoint
  7. Use torch.no_grad() for evaluation
  8. Clear gradients with optimizer.zerograd(setto_none=True) for efficiency

Source & license

This open-source skill is cataloged on AgentStack and links to its original source — we do not rehost the code.

Install and usage instructions live in the source repository linked above.

Reviews

No reviews yet — be the first.

Versions

  • v0.1.0 Imported from the upstream source.