Results for “psnr”
4 skillsTao Train Pose Classification
Train, evaluate, export, and run inference for pose classification models using ST-GCN on skeleton keypoint sequences.
2.2k · bundle
Pytorch Fsdp2
Adds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing. Use when models exceed single-GPU memory or when you need DTensor-based sharding with DeviceMesh.
10.4k · bundle
Rwkv Architecture
Use RWKV, a linear-time RNN-Transformer hybrid, for efficient long-context inference and training with constant memory usage.
10.4k · bundle
Pennylane
Train quantum circuits like neural networks with automatic differentiation, device-independent programming, and integration with PyTorch or JAX.
30.2k · bundle