model-code / clean /video /lipfd /options /test_options.py
deepsafe's picture
Add stripped inference-only model code mirror
9e14838 verified
Raw History Blame Contribute Delete
355 Bytes
from .base_options import BaseOptions
class TestOptions(BaseOptions):
def initialize(self, parser):
parser = BaseOptions.initialize(self, parser)
parser.add_argument('--model_path')
parser.add_argument('--eval', action='store_true', help='use eval mode during test time.')
self.isTrain = False
return parser