Max Stepanov mstepanov

panovr / finetune.py

Created March 2, 2017 23:04

Fine-tuning pre-trained models with PyTorch

	import argparse
	import os
	import shutil
	import time

	import torch
	import torch.nn as nn
	import torch.nn.parallel
	import torch.backends.cudnn as cudnn
	import torch.optim