Cannot assign a device for operation global_step: {{node global_step}} was explicitly assigned to /job:ps/task:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:GPU:0 ]. Make sure the device specification refers to a valid device.

[jingwei0109]

Describe the bug

Urgency

System information

• OS Platform and Distribution (e.g., Linux Ubuntu 18.04*):
• TensorFlow Version:
• Python version:
• ONNX version (if applicable, e.g. 1.11*):
• ONNXRuntime version (if applicable, e.g. 1.11*):

To Reproduce

Screenshots

Additional context

[fatcat-z]

Could you please share more details for an investigation? Like script, model file?