Skip to content

Commit

Permalink
Update infer_example.py
Browse files Browse the repository at this point in the history
  • Loading branch information
asarvi authored Aug 4, 2019
1 parent bc46c21 commit 91d79a5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion infer_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,5 +76,5 @@ class Option(object): pass
opt.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

inf = Inferer(opt)
t_probs = inf.evaluate(['happy memory', 'bad service', 'just normal food'])
t_probs = inf.evaluate(['happy memory', 'good service', 'just normal food'])
print(t_probs)

0 comments on commit 91d79a5

Please sign in to comment.