Skip to content

Commit

Permalink
[bug fix] match naming #153
Browse files Browse the repository at this point in the history
  • Loading branch information
voldemortX authored Aug 15, 2023
1 parent bf0827a commit 9a87153
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@


train = dict(
exp_name='resnet34_bezierlanenet_tusimple-aug2',
exp_name='resnet34_bezierlanenet_tusimple-aug1b',
workers=8,
batch_size=20,
checkpoint=None,
Expand All @@ -33,10 +33,10 @@
)

test = dict(
exp_name='resnet34_bezierlanenet_tusimple-aug2',
exp_name='resnet34_bezierlanenet_tusimple-aug1b',
workers=0,
batch_size=1,
checkpoint='./checkpoints/resnet34_bezierlanenet_tusimple-aug2/model.pt',
checkpoint='./checkpoints/resnet34_bezierlanenet_tusimple-aug1b/model.pt',
# Device args
device='cuda',

Expand Down

0 comments on commit 9a87153

Please sign in to comment.