작성
·
158
0
# train용 generator 생성, valid용 generator는 데이터 부족으로 위 args 설정에서 None으로 함.
train_gen,valid_gen = create_generators(args,b.preprocess_image)
# retinanet 기반 네트웍 모델 설정. weight값을 아직 설정하지 않았으며, args config 설정.
# model, training_model, prediction_model이 반환되나 이중 training_model만 사용
model, training_model, prediction_model = create_models(
backbone_retinanet=b.retinanet,
num_classes=train_gen.num_classes(),
weights=None,
multi_gpu=False,
freeze_backbone=True,
lr=1e-3,
config=args.config)
# callback 생성. epoch시 마다 발생하는 ModelCheckpoint, ReduceLROnPlateur callback 설정.
callbacks = create_callbacks(model, training_model, prediction_model, valid_gen,args)
해당코드를 돌리면 아래와같은 오류가 뜨는데요,, 이건 어떤 문제가 생긴건가요? args 이 reduce_lr_factor에 없다는거같은데요ㅡㅜ 어제 라쿤데이터셑에서 train.py부분 막힌거때문에 연달아 안되는것인지 해서요..
--------------------------------------------------------------------------- AttributeError Traceback (most recent call last) <ipython-input-16-642a2abb2662> in <module> 14 15 # callback 생성. epoch시 마다 발생하는 ModelCheckpoint, ReduceLROnPlateur callback 설정. ---> 16 callbacks = create_callbacks(model, training_model, prediction_model, valid_gen,args) ~/anaconda3/envs/tf115/lib/python3.6/site-packages/keras_retinanet-0.5.1-py3.6-linux-x86_64.egg/keras_retinanet/bin/train.py in create_callbacks(model, training_model, prediction_model, validation_generator, args) 194 callbacks.append(keras.callbacks.ReduceLROnPlateau( 195 monitor = 'loss', --> 196 factor = args.reduce_lr_factor, 197 patience = args.reduce_lr_patience, 198 verbose = 1, AttributeError: type object 'args' has no attribute 'reduce_lr_factor'
답변 1
0
class args: 에 맨 마지막에 아래를 추가해 주시면 됩니다.