인프런 커뮤니티 질문&답변
작성한 질문수
mask r-cnn custom data 학습 수행시 경고발생
작성
·
357
0
안녕하세요~~좋은 강의 잘 듣고 있습니다.
mask r-cnn강의를 듣고 custom data로 학습을 수행 시
1. /home/******/anaconda3/envs/tf113/lib/python3.6/site-packages/tensorflow/python/ops/gradients_impl.py:110: UserWarning: Converting sparse IndexedSlices to a dense Tensor of unknown shape. This may consume a large amount of memory.
"Converting sparse IndexedSlices to a dense Tensor of unknown shape. "
2. /home/****/anaconda3/envs/tf113/lib/python3.6/site-packages/keras/engine/training_generator.py:46: UserWarning: Using a generator with `use_multiprocessing=True` and multiple workers may duplicate your data. Please consider using the`keras.utils.Sequence class.
UserWarning('Using a generator with `use_multiprocessing=True`'
위 2개의 경고가 발생하고
Epoch 1/50 에서 계속 멈추었는지..느리게 진행되고 있는 건지...변화가 없어서 이렇게 문의 드립니다.
현재 gpu는 T4사용하고 있습니다. 감사합니다.
답변 1
0
안녕하십니까,
먼저 오류 내용으로만 보면 fit_generator( )호출 시 use_multiprocessing=True에서 과다 메모리 사용으로 문제가 있다고 나옵니다. 원래 keras의 use_multiprocessing=True는 좀 문제가 있지만 matter port mask_rnn 패키지에서는 default로 use_multiprocessing=True 로 적용되어도 별 문제가 없었습니다.
아래 사항 확인 부탁드립니다.
1. 수행 환경이 리눅스인지, tensorflow 환경이 tf113(tensorflow 1.13, keras 2.2) 인지 확인 부탁드립니다.
2. 주피터 노트북을 재 기동하거나, 커널은 다시 Restart 후에 수행 부탁드립니다. 다른 커널은 수행하지 않은 채로 해당 nucleus 주피터 노트북만 기동하시고 학습부탁드립니다.
감사합니다.