erfanasghariyan commited on
Commit
6d228e2
·
verified ·
1 Parent(s): 89c6652

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.json +9 -0
  2. model_architecture.py +6 -0
  3. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "vocab_size": 30522,
3
+ "embed_dim": 512,
4
+ "num_layers": 5,
5
+ "encoder_name": "microsoft/resnet-50",
6
+ "graph_window_size": 3,
7
+ "graph_steps": 3,
8
+ "dataset_id": "raidium/RadImageNet-VQA"
9
+ }
model_architecture.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ # Model Architecture Definition
2
+ import torch
3
+ import torch.nn as nn
4
+ import math
5
+
6
+ # GraphPropagationLayer
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b1a33b0220b72510d0776c99f4018fcccd739520057beaa803c14f0be99f4ce
3
+ size 291995043