X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=voxel_flow_train.py;h=dfe87b198e63a3286bb674361b7608b4509a74d6;hb=586b7753362bf4b02fec6490dcefe4073f1af8bd;hp=856ba857070fdd007f977eb7f7d76859f22987e0;hpb=bc1177902e6abfae7a65cf68b3172ba96c07ce05;p=voxel-flow diff --git a/voxel_flow_train.py b/voxel_flow_train.py index 856ba85..dfe87b1 100755 --- a/voxel_flow_train.py +++ b/voxel_flow_train.py @@ -52,7 +52,7 @@ def train(dataset_frame1, dataset_frame2, dataset_frame3): # Prepare model. model = Voxel_flow_model() - prediction = model.inference(input_placeholder) d + prediction = model.inference(input_placeholder) # reproduction_loss, prior_loss = model.loss(prediction, target_placeholder) reproduction_loss = model.loss(prediction, target_placeholder) # total_loss = reproduction_loss + prior_loss