nehalelkaref commited on
Commit
56443b7
·
1 Parent(s): 1d8d3a1

Update network.py

Browse files
Files changed (1) hide show
  1. network.py +3 -3
network.py CHANGED
@@ -6,9 +6,9 @@ from torch.nn.utils.rnn import pad_sequence
6
  from torch.nn.functional import cross_entropy, binary_cross_entropy
7
  from tqdm.auto import tqdm
8
 
9
- from .utils import Config, extract_spans, generate_targets
10
- from .representation import TransformerRepresentation
11
- from .layers import SpanEnumerationLayer
12
 
13
  DEFAULT_DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
 
 
6
  from torch.nn.functional import cross_entropy, binary_cross_entropy
7
  from tqdm.auto import tqdm
8
 
9
+ from utils import Config, extract_spans, generate_targets
10
+ from representation import TransformerRepresentation
11
+ from layers import SpanEnumerationLayer
12
 
13
  DEFAULT_DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14