Skip to content

Instantly share code, notes, and snippets.

@omarsar
Created October 14, 2018 00:53
Show Gist options
  • Save omarsar/319fafbac6d56b7b639069de2ab97bcd to your computer and use it in GitHub Desktop.
Save omarsar/319fafbac6d56b7b639069de2ab97bcd to your computer and use it in GitHub Desktop.
from collections import defaultdict
import time
import os
import numpy as np
from tqdm import tqdm
from tensorboardX import SummaryWriter
from medicaltorch import datasets as mt_datasets
from medicaltorch import models as mt_models
from medicaltorch import transforms as mt_transforms
from medicaltorch import losses as mt_losses
from medicaltorch import metrics as mt_metrics
from medicaltorch import filters as mt_filters
import torch
from torchvision import transforms
from torch.utils.data import DataLoader
from torch import autograd, optim
import torch.backends.cudnn as cudnn
import torch.nn as nn
import torchvision.utils as vutils
cudnn.benchmark = True
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment