Skip to content

Instantly share code, notes, and snippets.

View kirk86's full-sized avatar

kirk86

View GitHub Profile
@kevinzakka
kevinzakka / data_loader.py
Last active March 19, 2024 05:15
Train, Validation and Test Split for torchvision Datasets
"""
Create train, valid, test iterators for CIFAR-10 [1].
Easily extended to MNIST, CIFAR-100 and Imagenet.
[1]: https://discuss.pytorch.org/t/feedback-on-pytorch-for-kaggle-competitions/2252/4
"""
import torch
import numpy as np
@krischer
krischer / embarrassingly_parallel.py
Created June 21, 2015 09:38
Embarrassingly Parallel Code with MPI In Python
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Simple script illustrating how to perform embarrassingly parallel computations
in Python using MPI/mpi4py. I like this approach a lot as its very easy to get
right without having to deal with the complications arising from forked
processes which the multiprocessing module uses.
This script can be executed with or without `mpirun`; it will just run on one
core if not executed with it. With some more logic its also possible to make