gsar78 commited on
Commit
7f88877
1 Parent(s): 1c3979d

Update modeling_custom.py

Browse files
Files changed (1) hide show
  1. modeling_custom.py +1 -2
modeling_custom.py CHANGED
@@ -1,7 +1,6 @@
1
- # modeling_custom.py
2
  from transformers import XLMRobertaForSequenceClassification
3
  import torch.nn as nn
4
- import torch.nn.functional as F
5
 
6
  class CustomModel(XLMRobertaForSequenceClassification):
7
  def __init__(self, config, num_emotion_labels):
 
 
1
  from transformers import XLMRobertaForSequenceClassification
2
  import torch.nn as nn
3
+ import torch
4
 
5
  class CustomModel(XLMRobertaForSequenceClassification):
6
  def __init__(self, config, num_emotion_labels):