File size: 396 Bytes
b84549f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import os
class Recoverable:
def load_checkpoint(self):
pass
def save_checkpoint(self):
pass
def get_checkpoint_path(self):
ckp_path = os.getenv('NNI_CHECKPOINT_DIRECTORY')
if ckp_path is not None and os.path.isdir(ckp_path):
return ckp_path
return None
|