Python TrainsSaverの例

コード例 #1

ファイル: training.py プロジェクト: sambinkam/ignite-1

def get_save_handler(config):
    if exp_tracking.has_trains:
        from ignite.contrib.handlers.trains_logger import TrainsSaver

        return TrainsSaver(dirname=config.output_path.as_posix())

    return DiskSaver(config.output_path.as_posix())

コード例 #2

ファイル: main.py プロジェクト: lynphoenix/ignite

def get_save_handler(config):
    if config["with_trains"]:
        from ignite.contrib.handlers.trains_logger import TrainsSaver

        return TrainsSaver(dirname=config["output_path"])

    return DiskSaver(config["output_path"], require_empty=False)