Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
N
NiuTrans.Tensor
概览
Overview
Details
Activity
Cycle Analytics
版本库
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
问题
8
Issues
8
列表
Board
标记
里程碑
合并请求
0
Merge Requests
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
Snippets
成员
Collapse sidebar
Close sidebar
活动
图像
聊天
创建新问题
作业
提交
Issue Boards
Open sidebar
NiuTrans
NiuTrans.Tensor
Commits
383759ae
Commit
383759ae
authored
6 years ago
by
xiaotong
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
remove model dumping
parent
af3acf68
隐藏空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
4 行增加
和
4 行删除
+4
-4
source/sample/transformer/Transformer.cpp
+4
-4
没有找到文件。
source/sample/transformer/Transformer.cpp
查看文件 @
383759ae
...
...
@@ -83,7 +83,7 @@ int TransformerMain(int argc, const char ** argv)
model
.
InitModel
(
argc
,
args
);
isTrain
=
trainer
.
Train
(
trainFN
,
testFN
,
modelFN
,
&
model
);
model
.
Dump
(
fn1
);
//
model.Dump(fn1);
}
else
{
T2TModel
model
;
...
...
@@ -91,7 +91,7 @@ int TransformerMain(int argc, const char ** argv)
model
.
Read
(
fn
);
isTrain
=
trainer
.
Train
(
trainFN
,
testFN
,
modelFN
,
&
model
);
model
.
Dump
(
fn1
);
//
model.Dump(fn1);
}
if
(
trainer
.
useEpochCheckpoint
&&
strcmp
(
testFN
,
""
))
{
...
...
@@ -100,9 +100,9 @@ int TransformerMain(int argc, const char ** argv)
T2TModel
model
;
model
.
InitModel
(
argc
,
args
);
model
.
Read
(
fn1
);
//
model.Read(fn1);
tester
.
Test
(
testFN
,
fn2
,
&
model
);
//
tester.Test(testFN, fn2, &model);
}
if
(
!
isTrain
)
...
...
This diff is collapsed.
Click to expand it.
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论