Shamdan17 / pytorch_optimizationLinks
General training optimization for pytorch (multigpu usage, mixed precision training)
☆11Updated 2 years ago
Alternatives and similar repositories for pytorch_optimization
Users that are interested in pytorch_optimization are comparing it to the libraries listed below
Sorting:
- ☆30Updated 3 months ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,987Updated last year
- LaTeX template for scientific papers with some tips on structure, style, and language. Ready to go for IEEE and arXiv submissions. Best u…☆29Updated last year
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆1,539Updated last month
- [PAMI'23] TransFuser: Imitation with Transformer-Based Sensor Fusion for Autonomous Driving; [CVPR'21] Multi-Modal Fusion Transformer for…☆1,484Updated 2 months ago
- A unified interface to many trajectory forecasting datasets.☆433Updated 2 months ago
- PyTorch code for the paper "Model-Based Imitation Learning for Urban Driving".☆416Updated 2 years ago
- PyTorch implementation for the paper "Driving with LLMs: Fusing Object-Level Vector Modality for Explainable Autonomous Driving"☆568Updated last year
- [CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize …☆1,968Updated last year
- Deep Reinforcement Learning (PPO) in Autonomous Driving (Carla) [from scratch]☆520Updated last month
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,188Updated 2 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,292Updated last year
- MTR: Motion Transformer with Global Intention Localization and Local Movement Refinement, NeurIPS 2022.☆882Updated 2 years ago
- Scenic: A Jax Library for Computer Vision Research and Beyond☆3,743Updated this week
- Code for the METEOR dataset containing data loading scripts along with pre-trained models for 2D object detection and action-behavior pre…☆12Updated last year
- Download the NuPlan Dataset directly from the terminal☆48Updated 11 months ago
- The devkit of the nuPlan dataset.☆941Updated 4 months ago
- [ICCV 2021- Oral] Official PyTorch implementation for Generic Attention-model Explainability for Interpreting Bi-Modal and Encoder-Decode…☆888Updated 2 years ago
- Explainability for Vision Transformers☆1,053Updated 3 years ago
- Multimodaler Anomalie-Detektions Benchmark für simulierte Szenarien☆14Updated last year
- [ICCV'23] Hidden Biases of End-to-End Driving Models & A starter kit for the CARLA leaderboard 2.0.☆484Updated 2 weeks ago
- Extension of ScenarioNet to support Waymax/V-Max and GPUDrive format☆62Updated last month
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,955Updated last year
- Model interpretability and understanding for PyTorch☆5,517Updated this week
- ☆106Updated 10 months ago
- Simple implementation of OpenAI CLIP model in PyTorch.☆719Updated 2 months ago
- Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)☆1,979Updated last year
- [CoRL'22] PlanT: Explainable Planning Transformers via Object-Level Representations☆289Updated last month
- Implementation of CoCa, Contrastive Captioners are Image-Text Foundation Models, in Pytorch☆1,193Updated 2 years ago
- Joint Out-of-Distribution Detection and Uncertainty Estimation for Trajectory Prediction: Model, training and evaluation code.☆34Updated 5 months ago