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, |