复现memory in memory模型报错

我正在复现一个程序(MIM模型应用于moving_mnist)
环境为python3.6,tensorflow1.15,
代码地址:https://github.com/Yunbo426/MIM
在运行主程序时报错:
Fatal Python error: Aborted
Thread 0x

同时有警告:
2022-07-24 18:15:17.277398: W tensorflow/core/common_runtime/colocation_graph.cc:983] Failed to place the graph without changing the devices of some resources. Some of the operations (that had to be colocated with resource generating operations) are not supported on the resources' devices. Current candidate devices are [
/job:localhost/replica:0/task:0/device:CPU:0].
See below for details of this colocation group:
Colocation Debug Info:
Colocation group had the following types and supported devices:
Root Member(assigned_device_name_index_=-1 requested_device_name_='/device:GPU:0' assigned_device_name_='' resource_device_name_='/device:GPU:0' supported_device_types_=[CPU] possible_devices_=[]
Assign: CPU
Const: CPU
Switch: CPU
VariableV2: CPU
Identity: CPU
IsVariableInitialized: CPU
RefSwitch: CPU
Merge: CPU
AssignSub: CPU

Colocation members, user-requested devices, and framework assigned devices, if any:
predrnn/back_to_pixel/bias/Initializer/zeros (Const)
predrnn/back_to_pixel/bias (VariableV2) /device:GPU:0
predrnn/back_to_pixel/bias/Assign (Assign) /device:GPU:0
predrnn/back_to_pixel/bias/read (Identity) /device:GPU:0
IsVariableInitialized_117 (IsVariableInitialized) /device:GPU:0
cond_117/read/Switch (RefSwitch) /device:GPU:0
cond_117/Switch_1 (Switch)
predrnn/back_to_pixel/bias/ExponentialMovingAverage (VariableV2) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/IsVariableInitialized (IsVariableInitialized) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/Switch (Switch) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/switch_t (Identity) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/switch_f (Identity) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/pred_id (Identity) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/read/Switch (RefSwitch) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/read (Identity) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/Switch_1 (Switch)
predrnn/back_to_pixel/bias/ExponentialMovingAverage/cond/Merge (Merge) /device:GPU:0
cond_117/read/Switch_predrnn/back_to_pixel/bias/ExponentialMovingAverage (Switch) /device:GPU:0
cond_117/read_predrnn/back_to_pixel/bias/ExponentialMovingAverage (Identity) /device:GPU:0
cond_117/Merge_predrnn/back_to_pixel/bias/ExponentialMovingAverage (Merge) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/Assign (Assign) /device:GPU:0
predrnn/back_to_pixel/bias/ExponentialMovingAverage/read (Identity) /device:GPU:0
ExponentialMovingAverage/AssignMovingAvg_117 (AssignSub) /device:GPU:0
Assign_353 (Assign) /device:GPU:0
save/Assign_237 (Assign) /device:GPU:0
save/Assign_238 (Assign) /device:GPU:0

请问我该如何解决这个问题?
这个问题是如何产生的?

麻烦你把源码发给我一份我运行一下试试