File size: 105 Bytes
a25d5d4 |
1 2 3 4 5 6 7 |
import torch.nn as nn
from src.core import register
CrossEntropyLoss = register(nn.CrossEntropyLoss)
|
a25d5d4 |
1 2 3 4 5 6 7 |
import torch.nn as nn
from src.core import register
CrossEntropyLoss = register(nn.CrossEntropyLoss)
|