Skip to content

Latest commit

 

History

History
49 lines (34 loc) · 1.01 KB

running-local-machine.md

File metadata and controls

49 lines (34 loc) · 1.01 KB

Run the app on your local machine

CPU version

Training

$ export BACKEND=cpu 
$ export ACTION=train
$ ./runUberJar.sh --action ${ACTION} --output-model-dir .

Model file created is called CnnSentenceClassificationModel-cpu.pb.

Evaluating

$ export BACKEND=cpu 
$ export ACTION=evaluate
$ ./runUberJar.sh --action ${ACTION} \
                    --input-model-file CnnSentenceClassificationModel-${BACKEND}.pb

GPU version

Note: please ensure you have access to an Nvidia GPU and the necessary drivers i.e. CUDA, cuDNN etc... are installed.

Training

$ export BACKEND=cpu 
$ export ACTION=train
$ ./runUberJar.sh --action ${ACTION} --output-model-dir .

Model file created is called CnnSentenceClassificationModel-gpu.pb.

Evaluating

$ export BACKEND=gpu
$ export ACTION=evaluate
$ ./runUberJar.sh --action ${ACTION} \
                    --input-model-file CnnSentenceClassificationModel-${BACKEND}.pb

Return to main README.md