File size: 364 Bytes
bfabd59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
"""
Constants for the demo.
"""
import os
ONNX_MODEL_DIRECTORY = "demo/onnx-models"
LEELA_MODEL_DIRECTORY = "demo/leela-models"
FIGURE_DIRECTORY = "demo/figures"
ONNX_MODEL_NAMES = [
f for f in os.listdir(ONNX_MODEL_DIRECTORY)
if f.endswith(".onnx")
]
LEELA_MODEL_NAMES = [
f for f in os.listdir(LEELA_MODEL_DIRECTORY)
if f.endswith(".pb.gz")
]
|