0%
1
2
3
4
5
6
7
| a = Variable(torch.Tensor([1, 2, 3]))
print(a.size(0))
``
```python
if output.data.is_cuda:
output.data = output.data.cpu()
|
1
2
3
4
5
| keypoints = []
for sample_data in test_data_batches:
keypoints.append(output.numpy())
keypoints = np.concatenate(keypoints, axis=0)
|