summaryrefslogtreecommitdiff
path: root/training/dtrain/dtrain.cc
diff options
context:
space:
mode:
authorPatrick Simianer <p@simianer.de>2016-04-08 23:15:09 +0200
committerPatrick Simianer <p@simianer.de>2016-04-08 23:15:09 +0200
commit7e583a0880347caf4e9ec84c2c801d1b280cffdd (patch)
treefe83a3123182b72de1f15bfade4eb0c67e21f637 /training/dtrain/dtrain.cc
parent176f311f6b4b2048dd05e0304d66ae5c61a4506e (diff)
dtrain: fixes
Diffstat (limited to 'training/dtrain/dtrain.cc')
-rw-r--r--training/dtrain/dtrain.cc4
1 files changed, 2 insertions, 2 deletions
diff --git a/training/dtrain/dtrain.cc b/training/dtrain/dtrain.cc
index 53e8cd50..b488e661 100644
--- a/training/dtrain/dtrain.cc
+++ b/training/dtrain/dtrain.cc
@@ -173,10 +173,10 @@ main(int argc, char** argv)
SparseVector<weight_t> gradient_accum, update_accum;
if (use_adadelta && adadelta_input!="") {
vector<weight_t> grads_tmp;
- Weights::InitFromFile(adadelta_input+".gradient", &grads_tmp);
+ Weights::InitFromFile(adadelta_input+".gradient.gz", &grads_tmp);
Weights::InitSparseVector(grads_tmp, &gradient_accum);
vector<weight_t> update_tmp;
- Weights::InitFromFile(adadelta_input+".update", &update_tmp);
+ Weights::InitFromFile(adadelta_input+".update.gz", &update_tmp);
Weights::InitSparseVector(update_tmp, &update_accum);
}