(), learning_rate=learning_rate)\n", + " train_pqcnet = ms.nn.TrainOneStepCell(molecule_pqcnet, optimizer)\n", + " energy_diff = eps * 1000\n", + " energy_last = initial_energy.asnumpy() + energy_diff\n", + " iter_idx = 0\n", + " while abs(energy_diff) > eps:\n...