Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
pytorch_pretrained_bert
README.md
data_util.py
main.py
run.sh

README.md

Test Bert on CLOTH Dataset

This repository test the Bert model on the CLOTH Dataset. The framework of implementation is provided by the Pytorch Pretrained Bert repo.

The finetuned Bert model reach 86.0% accuracy on the test set, which is the same as the performance of Amazon Turkers. Leaderboard

Experiment Environment

The code is tested with Python3.6 and Pytorch 0.4.0

Code Usage

Download Dataset

You can require the dataset from our website, and decompress it in the root of this repo.

Preprocessing Data

First step is to preprocess data by running

python data_util.py

You should assign the name of the Bert model that you want to use in the file. It is "bert-large-uncased" by default.

Finetune Model

You can run the model with the best hyper-parameter setting by

./run.sh
You can’t perform that action at this time.