SimpleNet убивают во время вывода ⇐ Python

[Гость]

Я пытаюсь запустить сеть SimpleNet. После запуска я вижу, что он успешно тренирует дискриминатор. Но во время вывода его убивают, и я вижу вывод терминала, например:

Matplotlib создал каталог временного кэша в /tmp/matplotlib-rp7dlh3f, поскольку путь по умолчанию (/home/username/.config/matplotlib) не является каталогом, доступным для записи; настоятельно рекомендуется установить переменную среды MPLCONFIGDIR в каталог, доступный для записи, в частности, чтобы ускорить импорт Matplotlib и улучшить поддержку многопроцессорности. ИНФОРМАЦИЯ:__main__:аргументы командной строки: main.py --gpu 0 --seed 0 --log_group simplenet_mvtec --log_project MVTecAD_Results --results_path /home/username/ml_ai_cv_workspace/SimpleNet/results --run_name run net -b Wideesnet50 -le Layer2 -le Layer3 --pretrain_embed_dimension 1536 --target_embed_dimension 1536 --patchsize 3 --meta_epochs 40 --embedding_size 256 --gan_epochs 4 --noise_std 0.015 --dsc_hidden 1024 --dsc_layers 2 --dsc_margin .5 --pre_proj 1 набор данных --batch_size 8 --resize 329 --imagesize 288 -d винт -d таблетка -d капсула -d ковер -d сетка -d плитка -d дерево -d молния -d кабель -d зубная щетка -d транзистор -d metal_nut -d бутылка -d лесной орех -d кожа mvtec /home/имя пользователя/mvtech_dataset ИНФОРМАЦИЯ:__main__:Набор данных: train=320 test=160 ИНФОРМАЦИЯ:__main__:Набор данных: train=267 test=167 ИНФОРМАЦИЯ:__main__:Набор данных: train=219 test=132 ИНФОРМАЦИЯ:__main__:Набор данных: train=280 test=117 ИНФОРМАЦИЯ:__main__:Набор данных: train=264 test=78 ИНФОРМАЦИЯ:__main__:Набор данных: train=230 test=117 ИНФОРМАЦИЯ:__main__:Набор данных: train=247 test=79 ИНФОРМАЦИЯ:__main__:Набор данных: train=240 test=151 ИНФОРМАЦИЯ:__main__:Набор данных: train=224 test=150 ИНФОРМАЦИЯ:__main__:Набор данных: поезд=60 тест=42 ИНФОРМАЦИЯ:__main__:Набор данных: train=213 test=100 ИНФОРМАЦИЯ:__main__:Набор данных: train=220 test=115 ИНФОРМАЦИЯ:__main__:Набор данных: train=209 test=83 ИНФОРМАЦИЯ:__main__:Набор данных: train=391 test=110 ИНФОРМАЦИЯ:__main__:Набор данных: train=245 test=124 ИНФОРМАЦИЯ:__main__:Оценка набора данных [mvtec_screw] (1/15)... /home/username/.local/lib/python3.8/site-packages/torchvision/models/_utils.py:208: UserWarning: параметр pretrained устарел с версии 0.13 и может быть удален в будущем, используйте ' вместо этого весы. предупреждения.предупреждать( /home/username/.local/lib/python3.8/site-packages/torchvision/models/_utils.py:223: UserWarning: аргументы, отличные от перечисления веса или `None` для 'weights', устарели с версии 0.13 и могут быть удалены в будущем. Текущее поведение эквивалентно передаче Weights=Wide_ResNet50_2_Weights.IMAGENET1K_V1. Вы также можете использовать Weights=Wide_ResNet50_2_Weights.DEFAULT, чтобы получить самые актуальные веса. предупреждения.warn(сообщение) ИНФОРМАЦИЯ:__main__:Обучающие модели (1/1) ИНФОРМАЦИЯ:simplenet:Обучающий дискриминатор... эпоха:3 потеря:0.45132 lr:0.0002 p_true:0.45 p_fake:0.475: 100%|████████████████████████████| 4/4 [00:34