Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
49 lines (34 sloc) 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

You can’t perform that action at this time.