97d7fe68af9f69eac7fad77c7bf0d447f1abbb82,bootstrap/run.py,,run,#Any#,49

Before Change

    // display server name and GPU(s) id(s)
    Logger()(os.uname())
    if "CUDA_VISIBLE_DEVICES" in os.environ:
        Logger()("CUDA_VISIBLE_DEVICES="+os.environ["CUDA_VISIBLE_DEVICES"])

    // engine can train, eval, optimize the model
    // engine can save and load the model and optimizer

After Change

    Logger().log_dict("options", Options(), should_print=True) // display options
    Logger()(os.uname()) // display server name

    if torch.cuda.is_available():
        cudnn.benchmark = True
        Logger()("Available GPUs: {}".format(utils.available_gpu_ids()))

    // engine can train, eval, optimize the model
    // engine can save and load the model and optimizer
    engine = engines.factory()

    // dataset is a dictionary that contains all the needed datasets indexed by modes
    // (example: dataset.keys() -> ["train","eval"])

Instances