https://github.com/facebookresearch/pythia
Raw File
Tip revision: 79600ba27e5973615fd596a2fd10da65c9cf909a authored by sash on 30 June 2021, 18:34:45 UTC
[feat][PL][4/N] best ckpt
Tip revision: 79600ba
torchscript.py
# Copyright (c) Facebook, Inc. and its affiliates.

from typing import Dict, Optional

from torch import Tensor


def getattr_torchscriptable(
    dictionary: Dict[str, Tensor], key: str, default: Optional[Tensor] = None
) -> Optional[Tensor]:
    if key in dictionary:
        return dictionary[key]
    else:
        return default
back to top