1 repository
Awesome GitHub RepositoriesRuntime Shape Tracking
Capturing actual tensor dimensions during a forward pass using execution hooks.
Distinct from Tensor Shape Inferences: Distinct from Tensor Shape Inferences: focuses on recording runtime shapes via hooks rather than compile-time or static inference.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Runtime Shape Tracking. Refine with filters or upvote what's useful.
Awesome Runtime Shape Tracking GitHub Repositories
Găsește cele mai bune repo-uri cu AI.Vom căuta cele mai potrivite repository-uri folosind AI.
- sksq96/pytorch-summary
sksq96/pytorch-summary
4,054Vezi pe GitHubpytorch-summary este o colecție de utilitare pentru rețele neuronale PyTorch concepute pentru a genera rezumate ale modelelor, a calcula cerințele de memorie și a vizualiza formele tensorilor strat cu strat. Acesta funcționează ca un instrument de raportare care oferă defalcări detaliate ale straturilor rețelei și ale formelor de ieșire pentru a asista la debugging-ul și inspecția modelelor. Proiectul oferă capabilități specializate pentru estimarea utilizării totale a memoriei pentru forward și backward pass-uri pe baza dimensiunilor de intrare și a numărului de parametri. Generează vizualizări lizibile de către oameni ale structurilor modelelor pentru a verifica design-urile arhitecturale și a identifica nepotrivirile de dimensiuni între straturi. Instrumentul implementează analiza structurală prin parcurgerea recursivă a modulelor, urmărirea tensorilor bazată pe hook-uri și inferența formelor bazată pe input. Aceste capabilități permit agregarea numărului de parametri și maparea fluxului de date între operațiuni succesive.
Intercepts PyTorch forward passes using hooks to record the exact shape of tensors at every layer.
Pythondeep-learningkeraspytorch
