Skip to content

Commit 25be623

Browse files
authored
fix gpubox error (#1006)
* disable pir in all ps mode for Paddle 3.0 * fix gpubox error
1 parent a28e3f4 commit 25be623

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/static_gpubox_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
from __future__ import print_function
1516
import os
1617
os.environ['FLAGS_enable_pir_api'] = '0'
17-
from __future__ import print_function
1818
from utils.static_ps.reader_helper import get_reader, get_example_num, get_file_list, get_word_num
1919
from utils.static_ps.program_helper import get_model, get_strategy
2020
from utils.static_ps.common_ps import YamlHelper, is_distributed_env

0 commit comments

Comments
 (0)