multiple_futures_prediction/train_ngsim.py [35:42]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    if params.use_cuda:
      hist = hist.cuda()
      nbrs = nbrs.cuda()
      mask = mask.cuda()            
      fut = fut.cuda()
      mask = mask.cuda()
      if context is not None:
        context = context.cuda()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



multiple_futures_prediction/train_ngsim.py [233:240]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      if params.use_cuda:
        hist = hist.cuda()
        nbrs = nbrs.cuda()
        mask = mask.cuda()
        fut = fut.cuda()
        mask = mask.cuda()
        if context is not None:
          context = context.cuda()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -