편명장/님/(myeongjang.pyeon)
initial commit
287a683
raw
history blame contribute delete
641 Bytes
imports:
- $import os
- $import glob
- $import torch
- $import scripts
- $import scripts.inference
device: '$torch.device("cuda" if torch.cuda.is_available() else "cpu")'
model_config:
_target_: scripts.exaonepath.EXAONEPathV1Downstream
step_size: 256
patch_size: 256
macenko: true
device: '$@device'
model: '$@model_config'
input_dir: 'samples'
input_files: '$sorted(glob.glob(@input_dir+''/*.svs''))'
root_dir: '$os.path.dirname(os.path.dirname(scripts.__file__))'
inference:
- [email protected]_state_dict(torch.load(os.path.join(@root_dir, "models/exaonepath_v1.0.0_msi.pt")))
- $scripts.inference.infer(@model, @input_files)