search

xiangking / ark-nlp

A private nlp coding package, which quickly implements the SOTA solutions.
Apache License 2.0
310 stars 65 forks source link

更新task结构，并新增默认功能 #76

Closed xiangking closed 2 years ago

xiangking commented 2 years ago

PR types

Refactor

PR changes

更新task结构，并新增默认功能

Description

  1. 将SequenceClassificationTask的运行逻辑迁移到Task类中
  2. 增加save方法
  3. 修改加载optimizer的逻辑
  4. 新增多种scheduler并适配到task类
  5. 修改加载loss_function的逻辑