cvfinetune/examples/basic/main.py

#!/usr/bin/env python
if __name__ != '__main__': raise Exception("Do not import me!")

import socket
if socket.gethostname() != "sigma25":
	import matplotlib
	matplotlib.use('Agg')

import chainer
import logging

from chainer.training.updaters import StandardUpdater


from cvfinetune.finetuner import DefaultFinetuner
from cvfinetune.training.trainer import Trainer
from cvfinetune.dataset import BaseDataset
from cvfinetune.classifier import Classifier


from utils import parser

def main(args):
	if args.debug:
		chainer.set_debug(args.debug)
		logging.warning("DEBUG MODE ENABLED!")


	tuner = DefaultFinetuner(
		args,
		classifier_cls=Classifier,
		classifier_kwargs={},
		model_kwargs=dict(
			pooling=args.pooling,
		),

		dataset_cls=BaseDataset,

		updater_cls=StandardUpdater,
		updater_kwargs={},
	)


	tuner.run(trainer_cls=Trainer, opts=args)


main(parser.parse_args())