From a7ec5def7bbcbcc448b5e7fa1dcac8f8bac3ffad Mon Sep 17 00:00:00 2001
From: Wojciech Glogowski <glogowski@fb.com>
Date: Wed, 30 Aug 2017 12:37:06 -0700
Subject: [PATCH] data_parallel_model names fix

Summary: Updated usage of deprecated functions in data_parallel_model.py

Reviewed By: akyrola

Differential Revision: D5738512

fbshipit-source-id: a7767e518da777ece058bcad480e5df1d91e9b42
---
 caffe2/python/data_parallel_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/caffe2/python/data_parallel_model.py b/caffe2/python/data_parallel_model.py
index 17ae0b6aa2b..900d07ab8ae 100644
--- a/caffe2/python/data_parallel_model.py
+++ b/caffe2/python/data_parallel_model.py
@@ -779,9 +779,9 @@ def GetLearningRateBlobNames(model):
     '''
     if model._optimizer is not None:
         if model._device_type == caffe2_pb2.CPU:
-            return [model._optimizer.get_cpu_lr_blob_name()]
+            return [model._optimizer.get_cpu_blob_name('lr')]
         elif model._device_type == caffe2_pb2.CUDA:
-            return [model._optimizer.get_gpu_lr_blob_name(gpu)
+            return [model._optimizer.get_gpu_blob_name('lr', gpu)
                     for gpu in model._devices]
         else:
             raise Exception(