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