Commit bedf3c0c authored by hepj's avatar hepj
Browse files

修改README,增加训练脚本,完善模型转换代码

parent 49afe744
This diff is collapsed.
This diff is collapsed.
...@@ -14,9 +14,7 @@ ...@@ -14,9 +14,7 @@
import torch import torch
import argparse import argparse
#from modeling import BertForPreTraining, BertConfig from modeling import BertForPreTraining, BertConfig
from model import BertForPreTraining, BertConfig
def parse_arguments(): def parse_arguments():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
import torch
from .fused_gelu import bias_gelu_impl
__all__ = ["bias_gelu_impl"]
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment