https://github.com/facebookresearch/pythia
Tip revision: 79600ba27e5973615fd596a2fd10da65c9cf909a authored by sash on 30 June 2021, 18:34:45 UTC
[feat][PL][4/N] best ckpt
[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