                        dtype=torch.double).cuda()),
                requires_grad=True)
            input_fp = Parameter(
                torch.randn(
                    n_train_sample, channel_in, w_in, h_in,
                    dtype=torch.float).cuda(),