From b83c77e105a3e44d7242c0a2cbeb6ed32ff95bdd Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 12 Jan 2021 16:36:52 +0800 Subject: [PATCH] test(mge/test_save_load): remove test model file GitOrigin-RevId: 893b82b1c93b07e778310799b78b822e8eaa4246 --- imperative/python/test/integration/test_save_load.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/imperative/python/test/integration/test_save_load.py b/imperative/python/test/integration/test_save_load.py index e8075028..cfa65680 100644 --- a/imperative/python/test/integration/test_save_load.py +++ b/imperative/python/test/integration/test_save_load.py @@ -1,3 +1,12 @@ +# MegEngine is Licensed under the Apache License, Version 2.0 (the "License") +# +# Copyright (c) 2014-2020 Megvii Inc. All rights reserved. +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +import os + import numpy as np import megengine as mge @@ -53,6 +62,7 @@ def test_save_load(): optim = optimizer.SGD(net.parameters(), lr=1.0, momentum=0.9) optim.load_state_dict(checkpoint["opt_state"]) print("load done") + os.remove("simple.pkl") with gm: loss = net([1.23])