diff options
Diffstat (limited to 'training/dtrain/dtrain.cc')
| -rw-r--r-- | training/dtrain/dtrain.cc | 4 |
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); } |