dennispark
commited on
Commit
β’
bb24520
1
Parent(s):
0955864
Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ model = T5ForConditionalGeneration.from_pretrained('paust/pko-t5-large')
|
|
25 |
|
26 |
input_ids = tokenizer(["nsmc sentence: λΉμ μ μ΄λ¦μ 무μμΈκ°μ?"]).input_ids
|
27 |
labels = tokenizer(["T5 μ
λλ€."]).input_ids
|
28 |
-
outputs = model(input_ids, labels)
|
29 |
|
30 |
print(f"loss={outputs.loss} logits={outputs.logits}")
|
31 |
```
|
|
|
25 |
|
26 |
input_ids = tokenizer(["nsmc sentence: λΉμ μ μ΄λ¦μ 무μμΈκ°μ?"]).input_ids
|
27 |
labels = tokenizer(["T5 μ
λλ€."]).input_ids
|
28 |
+
outputs = model(input_ids=input_ids, labels=labels)
|
29 |
|
30 |
print(f"loss={outputs.loss} logits={outputs.logits}")
|
31 |
```
|