Skip to content

Commit 009745a

Browse files
authored
use_pretrained_model=1 for better accuracy (aws#879)
1 parent b577cfb commit 009745a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

sagemaker_neo_compilation_jobs/imageclassification_caltech/Image-classification-fulltraining-highlevel-neo.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,7 @@
191191
" epochs=5,\n",
192192
" learning_rate=0.01,\n",
193193
" top_k=2,\n",
194+
" use_pretrained_model=1,\n",
194195
" precision_dtype='float32')"
195196
]
196197
},

0 commit comments

Comments
 (0)