Spaces:
Running
Running
import torch as T | |
class Network(T.nn.Module): | |
def __init__(self): | |
super(Network, self).__init__() | |
def forward(self, image, segbox): | |
pass | |
def train_step(self, image, segment, criterion, segbox): | |
pass |