Function main

script/feature/efficientnet.py:273–282

test model

()

        return feature_maps, predict

def main():
  """
  test model
  """
  from torchsummary import summary
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
  feat_model = EfficientNetB3()
  # feat_model.load_state_dict(torch.load(''))
  feat_model.to(device)
  summary(feat_model, (3, 240, 427))

if __name__ == '__main__':
  main()