tinysleepnet
tinysleepnet copied to clipboard
Question on predict.py
I couldn't find the method evaluate_aug in predict.py However the training settings are "augment_seq": True, "augment_signal_full": True, test epochs run with "model.evaluate" instead of "model.evaluate_aug". Is there a specific reason?
if (config.get('augment_signal') is not None) and config['augment_signal']:
# Evaluate
test_outs = model.evaluate_aug(test_minibatch_fn)
else:
# Evaluate
test_outs = model.evaluate(test_minibatch_fn)