소스 검색

Fix typo in prompt-tuning-sst2.ipynb (#245)

Alexander Borzunov 2 년 전
부모
커밋
5367523df8
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      examples/prompt-tuning-sst2.ipynb

+ 1 - 1
examples/prompt-tuning-sst2.ipynb

@@ -399,7 +399,7 @@
     "    head_layer_position=HEAD_LAYER_POSITION,\n",
     ")\n",
     "cls_optimizer = AdamW(cls_model.parameters(), lr=LR, weight_decay=WEIGHT_DECAY)\n",
-    "cls_criterion = nn.CrossEntoryCriterion()\n",
+    "cls_criterion = nn.CrossEntropyCriterion()\n",
     "\n",
     "lr_scheduler = get_scheduler(\n",
     "    name=\"linear\", optimizer=cls_optimizer, num_warmup_steps=0, num_training_steps=len(train_dataloader)\n",