--correct_time_offset now has an indep argument to --time_offset, to …

…allow adjustment for various time shifts

inference.py

@@ -78,7 +78,7 @@
 parser.add_argument('--auxdata', action='store_true', help='inject auxillary data symbol')
 parser.add_argument('--txbpf', action='store_true', help='inject auxillary data symbol')
 parser.add_argument('--pilots2', action='store_true', help='insert pilot symbols inside z vectors, replacing data symbols')
-parser.add_argument('--correct_time_offset', action='store_true', help='correct phase shift caused by --time_offset (default off)')
+parser.add_argument('--correct_time_offset', type=int, default=0, help='correct phase shift caused by --correct_time_offset in samples (default off)')
 args = parser.parse_args()
 
 if len(args.h_file):

radae/radae.py

@@ -709,10 +709,9 @@ def forward(self, features, H, G=None):
 
         # genie based phase adjustment for time shift 
         if self.correct_time_offset:
-            print(rx_sym.shape)
-
+            print(self.correct_time_offset)
             # Use vector multiply to create a shape (batch,Nc) 2D tensor
-            phase_offset = -self.time_offset*torch.reshape(self.w,(1,self.Nc))
+            phase_offset = -self.correct_time_offset*torch.reshape(self.w,(1,self.Nc))
             phase_offset = torch.reshape(phase_offset,(num_batches,self.Nc,1))
 
             # change to (batch,Nc,timestep), as all time steps get the same phase offset