При переподключении к Колаборатории у меня завис компьютер

Question

Я пытался запустить свой код глубокого обучения в Colab Laboratory, но каждый раз, когда я выполняю этот прогресс на 26%, он автоматически отключается, и компьютер зависает при повторном подключении к серверу. Моя тренировка l oop показана ниже:

from tqdm import tqdm_notebook
import matplotlib
from PIL import Image
dev_gen_costs = []
config = tf.ConfigProto(device_count={'gpu':0})
config.gpu_options.allow_growth = True
sess = tf.InteractiveSession(config = config)
sess.run(tf.global_variables_initializer())
for iteration in tqdm_notebook(range(ITERS), desc='generate and discriminate'):
    time.sleep(0.01)
    start_time = time.time()
    num = iteration%(train_vis.shape[0]//BATCH_SIZE)
    data_IR = train_ir[num*BATCH_SIZE:(num+1)*BATCH_SIZE,:,:]
    data_VI = train_vis[num*BATCH_SIZE:(num+1)*BATCH_SIZE,:,:]
    data_IR = np.expand_dims(data_IR,axis = -1)
    data_VI = np.expand_dims(data_VI,axis = -1)
    feed_dict={real_data1_int:data_IR, real_data2_int:data_VI}
    # Train generator
    _ = sess.run([gen_train_op],feed_dict)
    # Train critic
    for i in range(CRITIC_ITERS):
        _costs, _ = sess.run([[disc_cost ,gen_cost],disc_train_op] ,feed_dict)
        dev_gen_costs.append(_costs)#loss

Answer 1

Cra sh из-за некоторых ошибок в коде, который вы запускаете.