def test_loadModel(self):
print(self.mode)
net = Model(self.mode, show_structure=False)
net.checkPoint("tm", self.epoch, "test_model")
net.loadModel("test_model/checkpoint/Model_tm_%d.pth" % self.epoch,
"test_model/checkpoint/Weights_tm_%d.pth" % self.epoch)
dir = "test_model/"
shutil.rmtree(dir)
def test_checkPoint(self):
net = Model(self.mode, show_structure=False)
net.checkPoint("tm", self.epoch, "test_model")
dir = "test_model/"
shutil.rmtree(dir)