Makefile

FEATURES=32
BATCHSIZE=8
MODEL=models/paintstorch2_guide_40/checkpoint_39.pth
GUIDE=--guide
BN=
CURRICULUM=

build:
	sudo docker build -t yliess86/paintstorch2:latest .
	sudo docker push yliess86/paintstorch2:latest

train:
	python3 -m kubeflow \
		--n_gpu 4 \
		--batch_size ${BATCHSIZE} \
		--features ${FEATURES} \
		${GUIDE} ${BN} ${CURRICULUM}

exp:
	python3 -m kubeflow \
		--n_gpu 4 \
		--batch_size ${BATCHSIZE} \
		--features ${FEATURES} \
		--guide --curriculum

	python3 -m kubeflow \
		--n_gpu 4 \
		--batch_size ${BATCHSIZE} \
		--features ${FEATURES} \
		--guide

	python3 -m kubeflow \
		--n_gpu 4 \
		--batch_size ${BATCHSIZE} \
		--features ${FEATURES}


	python3 -m kubeflow \
		--n_gpu 4 \
		--batch_size ${BATCHSIZE} \
		--features ${FEATURES} \
		--guide --bn --curriculum

	python3 -m kubeflow \
		--n_gpu 4 \
		--batch_size ${BATCHSIZE} \
		--features ${FEATURES} \
		--guide --bn
	
	python3 -m kubeflow \
		--n_gpu 4 \
		--batch_size ${BATCHSIZE} \
		--features ${FEATURES} \
		--bn

test:
	python -m evaluation.benchmark \
		--features ${FEATURES} \
		--batch_size ${BATCHSIZE} \
		--num_workers ${BATCHSIZE} \
		--dataset dataset \
		--model ${MODEL} \
		${BN}

convert:
	python -m evaluation.export \
		--features ${FEATURES} \
		--model ${MODEL} \
		--save docs/resources/paintstorch.onnx

	rm -rf docs/resources/paintstorch.onnx docs/resources/paintstorch.pb