fix argument

This commit is contained in:
lorenzo 2019-05-28 08:52:53 +02:00
parent 086a4d909a
commit bf7dcd6a3c

View File

@ -23,7 +23,7 @@ class RunKitti:
average_y = 0.48 average_y = 0.48
n_samples = 100 n_samples = 100
def __init__(self, model, dir_ann, dropout, hidden_size, n_stage, n_dropout, stereo): def __init__(self, model, dir_ann, dropout, hidden_size, n_stage, n_dropout, stereo=False):
# Set directories # Set directories
assert dir_ann, "Annotations folder is required" assert dir_ann, "Annotations folder is required"