chuuhtetnaing commited on
Commit
b99e9fe
·
1 Parent(s): 67569e0

remove np usage on int

Browse files
Files changed (1) hide show
  1. model/hrnetv2/hrnet_ocr.py +1 -1
model/hrnetv2/hrnet_ocr.py CHANGED
@@ -209,7 +209,7 @@ class HighResolutionNet(nn.Module):
209
  num_blocks=4 * [num_blocks], num_channels=num_channels)
210
 
211
  if self.ocr_on:
212
- last_inp_channels = np.int(np.sum(pre_stage_channels))
213
  ocr_mid_channels = 2 * ocr_width
214
  ocr_key_channels = ocr_width
215
 
 
209
  num_blocks=4 * [num_blocks], num_channels=num_channels)
210
 
211
  if self.ocr_on:
212
+ last_inp_channels = int(np.sum(pre_stage_channels))
213
  ocr_mid_channels = 2 * ocr_width
214
  ocr_key_channels = ocr_width
215