MHasanUnical commited on
Commit
b213c22
·
verified ·
1 Parent(s): d23a0dc

Upload Model.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. Model.py +1 -0
Model.py CHANGED
@@ -4,6 +4,7 @@ import torch.nn as nn
4
  import torchvision
5
  from torchvision import transforms
6
  import torch.nn.functional as F
 
7
  from transformers import PreTrainedModel
8
  ######################################################################
9
  # IMAGE DOWN SAMPLING
 
4
  import torchvision
5
  from torchvision import transforms
6
  import torch.nn.functional as F
7
+ from ModelConfiguration import VesselSegmentConfig
8
  from transformers import PreTrainedModel
9
  ######################################################################
10
  # IMAGE DOWN SAMPLING