Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 187 Bytes
416833a |
1 2 3 4 5 6 7 8 9 10 11 12 |
task: multiclass
num_classes: 3
out:
_target_: torch.nn.LazyLinear
out_features: ${num_classes}
loss:
_target_: torch.nn.CrossEntropyLoss
activation:
_target_: torch.nn.Softmax |