sender.py: add --debug

Author: francisyyan

dagger/run_sender.py

@@ -62,9 +62,10 @@ def sample_action(self, state):
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument('port', type=int)
+    parser.add_argument('--debug', action='store_true')
     args = parser.parse_args()
 
-    sender = Sender(args.port)
+    sender = Sender(args.port, debug=args.debug)
 
     model_path = path.join(project_root.DIR, 'dagger', 'model', 'model')
 

env/.gitignore

@@ -1,2 +1,3 @@
 *.trace
 !12mbps.trace
+sampling_time

env/sender.py

@@ -30,8 +30,9 @@ class Sender(object):
     action_mapping = format_actions(["/2.0", "-10.0", "+0.0", "+10.0", "*2.0"])
     action_cnt = len(action_mapping)
 
-    def __init__(self, port=0, train=False):
+    def __init__(self, port=0, train=False, debug=False):
         self.train = train
+        self.debug = debug
 
         # UDP socket and poller
         self.peer_addr = None
@@ -47,8 +48,9 @@ def __init__(self, port=0, train=False):
 
         self.dummy_payload = 'x' * 1400
 
-        self.sampling_file = open(path.join(project_root.DIR, 'env', 'sampling_time'), 'w', 0)
-        
+        if self.debug:
+            self.sampling_file = open(path.join(project_root.DIR, 'env', 'sampling_time'), 'w', 0)
+
         # congestion control related
         self.seq_num = 0
         self.next_ack = 0
@@ -75,7 +77,8 @@ def __init__(self, port=0, train=False):
             self.rtt_buf = []
 
     def cleanup(self):
-        self.sampling_file.close()
+        if self.debug and self.sampling_file:
+            self.sampling_file.close()
         self.sock.close()
 
     def handshake(self):
@@ -186,9 +189,13 @@ def recv(self):
                      self.cwnd]
 
             # time how long it takes to get an action from the NN
-            start_sample = time.time()
+            if self.debug:
+                start_sample = time.time()
+
             action = self.sample_action(state)
-            self.sampling_file.write('%.2f ms\n' % ((time.time() - start_sample) * 1000))
+
+            if self.debug:
+                self.sampling_file.write('%.2f ms\n' % ((time.time() - start_sample) * 1000))
 
             self.take_action(action)