diff --git a/save_model.py b/save_model.py index de102c7..921d227 100644 --- a/save_model.py +++ b/save_model.py @@ -1,4 +1,3 @@ -#!/usr/bin/env python3 #!/usr/bin/env python """Saves the model as a TorchScript. @@ -14,7 +13,6 @@ parameters before saving it and it will gzip down to 780K: for p in model.parameters(): p.detach().zero_() """ - import glob import os import sys