An implementation of Efficientnet in PyTorch
Pytorch
B0:
""" | |
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. | |
""" |
extern crate vulkano; | |
extern crate vulkano_shaders; | |
use vulkano::{ | |
sync, | |
buffer::{BufferUsage, CpuAccessibleBuffer}, | |
command_buffer::{AutoCommandBufferBuilder, CommandBufferUsage}, | |
device::{ | |
physical::{PhysicalDevice}, | |
QueueCreateInfo, |