INNER CODE UNIT · Python

build_feature_extractor

GaParmar/clean-fid · cleanfid/features.py:60

def build_feature_extractor(mode, device=torch.device("cuda"), use_dataparallel=True):
    if mode == "legacy_pytorch":
        feat_model = feature_extractor(
            name="pytorch_inception",
            resize_inside=False,
            device=device,
            use_dataparallel=use_dataparallel,
        )
    elif mode == "legacy_tensorflow":
        feat_model = feature_extractor(
            name="torchscript_inception",
            resize_inside=True,
            device=device,
            use_dataparallel=use_dataparallel,
        )
    elif mode == "clean":
        feat_model = feature_extractor(
            name="torchscript_inception",

View source record →