выполнение моделей Keras CNN продолжает сбивать мой компьютер

Question

Я пытаюсь углубиться в изучение R. Используя различные блоги в Интернете, я пытаюсь проверить их код и посмотреть, как они на самом деле работают. С керасом я не уверен почему, но каждый раз, когда я запускаю модель, она продолжает падать. Извините, если я не предоставил достаточно информации. Я использую кодовый раздел AMD GPU и пример процессора

history <- model %>% fit_generator(
  train_generator,
  steps_per_epoch = 100,
  epochs = 100,
  validation_data = validation_generator,
  validation_steps = 50,
)
use_multiprocessing=False

также

hist <- model %>% fit_generator(
  # training data
  train_image_array_gen,

  # epochs
  steps_per_epoch = as.integer(train_samples / batch_size), 
  epochs = epochs, 

  # validation data
  validation_data = valid_image_array_gen,
  validation_steps = as.integer(valid_samples / batch_size),

  # print progress
  verbose = 2,
  callbacks = list(
    # save best model after every epoch
    callback_model_checkpoint("C:/Users/My Account/Desktop/fruits_checkpoints.h5", save_best_only = TRUE),

  )

Answer 1

Похоже, проблема в том, что Keras использует tensorflow-gpu. Попробуйте запустить модель после установки tensorflow версии процессора. Так как вы используете AMD gpu, вы не сможете использовать версию tenorflow gpu с библиотеками cudnn.