Skip to content

Commit e4a0ecf

Browse files
committed
wip pytorch
1 parent b9e032c commit e4a0ecf

3 files changed

Lines changed: 1 addition & 0 deletions

File tree

172 Bytes
Binary file not shown.
Binary file not shown.

sdk/diffgram/pytorch_diffgram/diffgram_pytorch_dataset.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ def __process_instance(self, instance):
2929
if instance['type'] == 'box':
3030
result = np.array([instance['x_min'], instance['y_min'], instance['x_max'], instance['y_max']])
3131
result = torch.tensor(result)
32+
3233
return result
3334

3435
def __len__(self):

0 commit comments

Comments
 (0)