При успешном обучении сети vgg19 файл контрольных точек записывается на локальный диск, а затем загружается в НОВУЮ модель для проверки. Последняя операция завершается с ошибкой:
RuntimeError: CUDA не хватает памяти. Попытка выделить 392,00 МБ (GPU 0; общая емкость 6,00 ГБ; 3,71 ГБ уже выделено; 144,13 МБ свободного; 4,16 ГБ, зарезервировано PyTorch в целом)
Сборка мусора НЕ помогла; Сокращение входного размера пакета оказалось жизнеспособным обходным решением на НЕКОТОРЫХ моделях, но не на всех, например, если я использую vgg16_bn, исключение, кажется, происходит независимо. Здесь Подробнее