Skip to content

Instantly share code, notes, and snippets.

View programming-datascience's full-sized avatar

Programming, Datascience, Others programming-datascience

View GitHub Profile
@programming-datascience
programming-datascience / pytorch-m1-gpu-mnist.py
Created May 21, 2022 19:13
Training MNIST on the M1 GPU with PyTorch.
"""
MNIST with PyTorch on Apple Silicon GPU
Script will be linked in the description as a Github Gist.
Install PyTorch nightly with this command:
pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
Code borrowed from PyTorch Examples.
"""
@SharanSMenon
SharanSMenon / vulkano_vector_scalar_product.rs
Created March 23, 2022 01:12
Vulkano 0.29 Vector-Scalar Product program. Multiplies a vector of 65K numbers by 12 using the GPU device. Vulkano and Vulkano shaders required for this.
extern crate vulkano;
extern crate vulkano_shaders;
use vulkano::{
sync,
buffer::{BufferUsage, CpuAccessibleBuffer},
command_buffer::{AutoCommandBufferBuilder, CommandBufferUsage},
device::{
physical::{PhysicalDevice},
QueueCreateInfo,
@SharanSMenon
SharanSMenon / efficientnet-pytorch.md
Last active November 10, 2023 04:00
EfficientNet Implementation PyTorch

EfficientNet Implementation

An implementation of Efficientnet in PyTorch

Results

Pytorch

B0:

@SharanSMenon
SharanSMenon / retinanet.ipynb
Created January 8, 2022 20:46
An implementation of Retinanet.
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.