Skip to content
#

pytorch-implementation

Star

Here are 1,802 public repositories matching this topic...

Language: All
Filter by language
All 1,802 Python 1,118 Jupyter Notebook 594 C++ 11 HTML 9 Kotlin 5 CSS 3 Roff 3 JavaScript 3 C# 2 C 1
Sort: Most stars
Sort options
Most stars Fewest stars Most forks Fewest forks Recently updated Least recently updated
pytorch-GAT

gordicaleksa / pytorch-GAT

Star 2.5k

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

python deep-learning jupyter pytorch attention attention-mechanism graph-attention-networks self-attention pytorch-implementation gat graph-attention-network pytorch-gat gat-tutorial
  • Updated Nov 17, 2022
  • Jupyter Notebook

MIC-DKFZ / medicaldetectiontoolkit

Star 1.3k

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

deep-neural-networks deep-learning detection medical-imaging segmentation object-detection medical-image-computing semantic-segmentation 3d-models medical-image-processing mask-rcnn u-net medical-image-analysis pytorch-cnn retina-net 3d-object-detection pytorch-implementation pytorch-deeplearning retina-unet 3d-mask-rcnn
  • Updated Jun 17, 2024
  • Python

LibCity / Bigscity-LibCity

Star 935

LibCity: An Open Library for Urban Spatial-temporal Data Mining

deep-learning toolkit traffic eta map-matching representation-learning on-demand-service spatio-temporal traffic-prediction trajectory-prediction time-series-prediction spatio-temporal-prediction traffic-flow-prediction pytorch-implementation od-matrix traffic-forecasting estimated-time-of-arrival traffic-accident-prediction traffic-speed-prediction libcity
  • Updated Nov 21, 2024
  • Python

Improve this page

Add a description, image, and links to the pytorch-implementation topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the pytorch-implementation topic, visit your repo's landing page and select "manage topics."

Learn more