A computer lab, because I make images, with pycharm ran tensorflow - the gpu version of the code, but just "train" (to run the training set, training the two pictures) is automatically shutdown restart, help, and after the restart, the function can be used normally as usual