NSMC classification example

Run on a GPU machine

• build a docker image

  docker build -t kobart -f docker/Dockerfile .

• run a docker container

  cd ~/KoBART # root directory of this repository
  docker run --gpus '"device=0"' --rm -it \
      -v $HOME/KoBART:$HOME/KoBART \
      -e PYTHONPATH="$HOME/KoBART" \
      -w "$HOME/KoBART/examples" \
      --name "kobart" \
      kobart /bin/sh

• finetune KoBART model with NSMC

  • ⚠️ run on the docker container

  • finetune

    python nsmc.py --gpus 1 --max_epochs 3 --default_root_dir .cache --gradient_clip_val 1.0