(self, data, batch_size)
| 79 | return loss, result |
| 80 | |
| 81 | def infer_completion(self, data, batch_size): |
| 82 | bev_seq = data['bev_seq'] |
| 83 | trans_matrices = data['trans_matrices'] |
| 84 | num_agent = data['num_agent'] |
| 85 | |
| 86 | result, ind_pred = self.model(bev_seq, trans_matrices, num_agent, batch_size=batch_size) |
| 87 | |
| 88 | return result |
| 89 | |
| 90 | # Used my MAE model in scene completion, added by Juexiao |
| 91 | def step_mae_completion(self, data, batch_size, mask_ratio, loss_fn='mse', trainable = False): |
nothing calls this directly
no outgoing calls
no test coverage detected