Skip to content

Commit

Permalink
Update train_student.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
HeegonJin authored Mar 19, 2024
1 parent 3eac3ea commit 0638c16
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions train_student.sh
Original file line number Diff line number Diff line change
@@ -1,9 +1,6 @@
# SPDX-FileCopyrightText: Ⓒ 2024 NCSOFT Corporation. All Rights Reserved.
# SPDX-License-Identifier: BSD-3-Clause

# SPDX-FileCopyrightText: Ⓒ 2024 NCSOFT Corporation. All Rights Reserved.
# SPDX-License-Identifier: BSD-3-Clause

#!/bin/bash
base_dir=$path_to_fairseq
export PYTHONPATH="${PYTHONPATH}:$base_dir"
Expand Down Expand Up @@ -33,4 +30,4 @@ CUDA_VISIBLE_DEVICES=$GPU_NUM fairseq-train $data_dir/$data \
--eval-bleu-print-samples \
--best-checkpoint-metric bleu --maximize-best-checkpoint-metric \
--keep-last-epochs 2 --patience 10 --max-epoch 100 --save-dir $data_dir/$student_model/$data \
--user-dir $custom_model_dir | tee -a $data_dir/$student_model/$data/train.log
--user-dir $custom_model_dir | tee -a $data_dir/$student_model/$data/train.log

0 comments on commit 0638c16

Please sign in to comment.