File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -603,6 +603,7 @@ We recommend using an `argument parser <https://docs.python.org/3.5/howto/argpar
603
603
Using the ``argparse `` library as an example, this part of the code would look something like this:
604
604
605
605
.. code :: python
606
+
606
607
parser = argparse.ArgumentParser()
607
608
608
609
# hyperparameters sent by the client are passed as command-line arguments to the script.
@@ -624,6 +625,7 @@ Note now that saving the model will not be done by default; this must be done by
624
625
If you were previously relying on the default save method, here is one you can copy into your code:
625
626
626
627
.. code :: python
628
+
627
629
def save (model_dir , model ):
628
630
model.symbol.save(os.path.join(model_dir, ' model-symbol.json' ))
629
631
model.save_params(os.path.join(model_dir, ' model-0000.params' ))
You can’t perform that action at this time.
0 commit comments