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 chainer_addons.models.classifier import Classifier

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())