restore_code (PaddlePaddle#323)

graphcore · Dec 13, 2021 · aa014d4 · aa014d4
1 parent b211713
commit aa014d4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/paddle/fluid/framework/ipu/ipu_executor.cc b/paddle/fluid/framework/ipu/ipu_executor.cc
@@ -52,9 +52,7 @@ void Executor::Prepare(const std::string &proto) {
   session_->prepareDevice();
   VLOG(10) << "Preparing session device...done";
 
-  if (ipu_strategy_->is_training) {
-    SetWeightsIO();
-  }
+  SetWeightsIO();
 
   VLOG(10) << "Copy weights from paddle to popart...";
   WeightsFromPaddle();