(self, ps_rref)
| 70 | class Trainer(object): |
| 71 | |
| 72 | def __init__(self, ps_rref): |
| 73 | self.ps_rref = ps_rref |
| 74 | self.loss_fn = nn.MSELoss() |
| 75 | self.one_hot_indices = torch.LongTensor(batch_size) \ |
| 76 | .random_(0, num_classes) \ |
| 77 | .view(batch_size, 1) |
| 78 | |
| 79 | def get_next_batch(self): |
| 80 | for _ in range(num_batches): |
nothing calls this directly
no outgoing calls
no test coverage detected