Repository navigation

#

pytorch-tensors

PyTorch tutorial notebooks for beginners.

Jupyter Notebook
6
2 年前

Optimizing PyTorch Model Training by Wrapping Memory Mapped Tensors on Nvidia GPUs with TensorDict.

Python
1
3 个月前

Evaluation of multiple graph neural network models—GCN, GAT, GraphSAGE, MPNN and DGI—for node classification on graph-structured data. Preprocessing includes feature normalization and adjacency-matrix regularization, and an ensemble of model predictions boosts performance. The best ensemble achieves 83.47% test accuracy.

Jupyter Notebook
1
3 个月前

PyTorch is a popular open-source machine learning library that provides tools for building and training neural networks. One of the key features of PyTorch is its ability to automatically compute gradients for tensor operations, which is essential for efficient training of neural networks.

Jupyter Notebook
0
2 年前

Optimizing PyTorch Model Training by Wrapping Memory Mapped Tensors on an Nvidia GPU with TensorDict.

Python
0
13 分钟前