Skip to content

Instantly share code, notes, and snippets.

@giacaglia
Created December 8, 2019 00:53
Show Gist options
  • Save giacaglia/48e561e3c9f3cca6b8289aefc9dd0e95 to your computer and use it in GitHub Desktop.
Save giacaglia/48e561e3c9f3cca6b8289aefc9dd0e95 to your computer and use it in GitHub Desktop.
import os
from datetime import datetime
import argparse
import torch.multiprocessing as mp
import torchvision
import torchvision.transforms as transforms
import torch
import torch.nn as nn
import torch.distributed as dist
from apex.parallel import DistributedDataParallel as DDP
from apex import amp
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment