Method DumpParam

paddle/fluid/framework/device_worker.cc:245–272

}

void DeviceWorker::DumpParam(const Scope& scope, const int batch_id) {
  std::ostringstream os;
  int device_id =
      static_cast<int>(static_cast<unsigned char>(place_.GetDeviceId()));
  for (auto& param : *dump_param_) {
    os.str("");
    Variable* var = scope.FindVar(param);
    if (var == nullptr || !var->IsInitialized()) {
      continue;
    }
    if (!var->IsType<DenseTensor>()) {
      continue;
    }
    phi::DenseTensor* tensor = var->GetMutable<DenseTensor>();
    if (tensor == nullptr || !tensor->IsInitialized()) {
      continue;
    }
    DenseTensor cpu_tensor;
    if (phi::is_gpu_place(tensor->place())) {
      TensorCopySync(*tensor, CPUPlace(), &cpu_tensor);
      tensor = &cpu_tensor;
    }
    int64_t len = tensor->numel();
    os << "(" << device_id << "," << batch_id << "," << param << ")"
       << PrintDenseTensor(tensor, 0, len);
    writer_ << os.str();
  }
}

void DeviceWorker::InitRandomDumpConfig(const TrainerDesc& desc) {
  bool is_dump_in_simple_mode = desc.is_dump_in_simple_mode();