Update main.py

main.py

@@ -113,7 +113,7 @@ if __name__ == '__main__':
             prob = CONFIG.DATA.EVAL.transition_probs[0]
             loss_percent = (1 - prob[0]) / (2 - prob[0] - prob[1]) * 100
             print('Evaluate with real trace' if masking == 'real' else
-                  'Evaluate with generated trace with {:.2f}% packet loss'.format(prob))
+                  'Evaluate with generated trace with {:.2f}% packet loss'.format(loss_percent))
         elif args.mode == 'test':
             model.cuda(device=0)
             testset = BlindTestLoader(test_dir=CONFIG.TEST.in_dir)