kamangir commited on
Commit ·
845b45b
1
Parent(s): 02c423d
validating fashion_mnist train - kamangir/bolt#689
Browse files- abcli/fashion_mnist.sh +7 -4
- abcli/image_classifier.sh +3 -1
- image_classifier/__init__.py +1 -1
abcli/fashion_mnist.sh
CHANGED
|
@@ -12,8 +12,8 @@ function abcli_fashion_mnist() {
|
|
| 12 |
"ingest fashion_mnist data."
|
| 13 |
abcli_help_line "fashion_mnist predict object_1" \
|
| 14 |
"run fashion_mnist model object_1 predict."
|
| 15 |
-
abcli_help_line "fashion_mnist train" \
|
| 16 |
-
"train fashion_mnist."
|
| 17 |
|
| 18 |
if [ "$(abcli_keyword_is $2 verbose)" == true ] ; then
|
| 19 |
python3 -m fashion_mnist --help
|
|
@@ -38,6 +38,9 @@ function abcli_fashion_mnist() {
|
|
| 38 |
if [ "$task" == "train" ] ; then
|
| 39 |
local model_object=$abcli_object_name
|
| 40 |
|
|
|
|
|
|
|
|
|
|
| 41 |
abcli_select - ~trail
|
| 42 |
abcli_fashion_mnist ingest
|
| 43 |
local data_object=$abcli_object_name
|
|
@@ -45,8 +48,8 @@ function abcli_fashion_mnist() {
|
|
| 45 |
abcli_select $model_object ~trail
|
| 46 |
abcli_image_classifier_train \
|
| 47 |
$data_object \
|
| 48 |
-
~color,~convnet \
|
| 49 |
-
${@:
|
| 50 |
|
| 51 |
return
|
| 52 |
fi
|
|
|
|
| 12 |
"ingest fashion_mnist data."
|
| 13 |
abcli_help_line "fashion_mnist predict object_1" \
|
| 14 |
"run fashion_mnist model object_1 predict."
|
| 15 |
+
abcli_help_line "fashion_mnist train [validate]" \
|
| 16 |
+
"train fashion_mnist [for validation]."
|
| 17 |
|
| 18 |
if [ "$(abcli_keyword_is $2 verbose)" == true ] ; then
|
| 19 |
python3 -m fashion_mnist --help
|
|
|
|
| 38 |
if [ "$task" == "train" ] ; then
|
| 39 |
local model_object=$abcli_object_name
|
| 40 |
|
| 41 |
+
local options=$2
|
| 42 |
+
local do_validate=$(abcli_option_get_unpacked "$options" "validate" 0)
|
| 43 |
+
|
| 44 |
abcli_select - ~trail
|
| 45 |
abcli_fashion_mnist ingest
|
| 46 |
local data_object=$abcli_object_name
|
|
|
|
| 48 |
abcli_select $model_object ~trail
|
| 49 |
abcli_image_classifier_train \
|
| 50 |
$data_object \
|
| 51 |
+
~color,~convnet,validate=$do_validate \
|
| 52 |
+
${@:3}
|
| 53 |
|
| 54 |
return
|
| 55 |
fi
|
abcli/image_classifier.sh
CHANGED
|
@@ -84,10 +84,12 @@ function abcli_image_classifier_train() {
|
|
| 84 |
local do_validate=$(abcli_option_int "$options" "validate" 0)
|
| 85 |
|
| 86 |
local extra_args=""
|
| 87 |
-
if [ "$do_validate" ==
|
| 88 |
local extra_args="--epochs 2"
|
| 89 |
fi
|
| 90 |
|
|
|
|
|
|
|
| 91 |
python3 -m image_classifier \
|
| 92 |
train \
|
| 93 |
--color $do_color \
|
|
|
|
| 84 |
local do_validate=$(abcli_option_int "$options" "validate" 0)
|
| 85 |
|
| 86 |
local extra_args=""
|
| 87 |
+
if [ "$do_validate" == 1 ] ; then
|
| 88 |
local extra_args="--epochs 2"
|
| 89 |
fi
|
| 90 |
|
| 91 |
+
abcli_log "image_classifier.train($data_object): $options"
|
| 92 |
+
|
| 93 |
python3 -m image_classifier \
|
| 94 |
train \
|
| 95 |
--color $do_color \
|
image_classifier/__init__.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
name = "image_classifier"
|
| 2 |
|
| 3 |
-
version = "1.1.
|
| 4 |
|
| 5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
|
|
|
| 1 |
name = "image_classifier"
|
| 2 |
|
| 3 |
+
version = "1.1.57"
|
| 4 |
|
| 5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|