Thanks for your response. My multi-label classification model is as follows:
classifier = nn.Sequential(
nn.Conv1d(128, 128, 1),
nn.ReLU(),
nn.BatchNorm1d(128),
nn.Dropout(p=0.2),
nn.Conv1d(128, 128, 1),
nn.ReLU(),
nn.BatchNorm1d(128),
nn.Dropout(p=0.2),
nn.Conv1d(128, 1, 1)
)
I’m making use of BCEWithLogitsLoss
and I describe how I’m using it in this post: Loss for Multi-label Classifier
Please let me know how I can improve it.