MCPcopy Create free account
hub / github.com/FreeformRobotics/OTS / __init__

Method __init__

lib/utils/data/dataset.py:37–40  ·  view source on GitHub ↗
(self, data_tensor, target_tensor)

Source from the content-addressed store, hash-verified

35 """
36
37 def __init__(self, data_tensor, target_tensor):
38 assert data_tensor.size(0) == target_tensor.size(0)
39 self.data_tensor = data_tensor
40 self.target_tensor = target_tensor
41
42 def __getitem__(self, index):
43 return self.data_tensor[index], self.target_tensor[index]

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected