tinysleepnet icon indicating copy to clipboard operation
tinysleepnet copied to clipboard

Question on predict.py

Open sajjadkarimi91 opened this issue 3 years ago • 0 comments

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)

sajjadkarimi91 avatar Dec 04 '21 15:12 sajjadkarimi91