├── data ├── __init__.py └── pipe.py ├── model ├── __init__.py ├── bart.py ├── callbacks.py ├── generater.py ├── losses.py ├── metrics.py ├── modeing_bart.py └── utils.py ├── predictor.py ├── readme.md ├── requirements.txt └── train.py /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/pipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/data/pipe.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/bart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/model/bart.py -------------------------------------------------------------------------------- /model/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/model/callbacks.py -------------------------------------------------------------------------------- /model/generater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/model/generater.py -------------------------------------------------------------------------------- /model/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/model/losses.py -------------------------------------------------------------------------------- /model/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/model/metrics.py -------------------------------------------------------------------------------- /model/modeing_bart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/model/modeing_bart.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/model/utils.py -------------------------------------------------------------------------------- /predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/predictor.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | transformers==3.4.0 2 | pytorch==1.7.1 -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhcc/BARTNER/HEAD/train.py --------------------------------------------------------------------------------