Merge pull request #8 from metrics-lab/main

updates codebase

.gitignore

@@ -0,0 +1,7 @@
+utils/__pycache__/*
+outputs/*
+models/MSG3D/__pycache__/*
+logs/*
+models/__pycache__/*
+data/*
+*test*

config/SiT/training/hparams.yml

@@ -17,7 +17,7 @@ training:
   LR: 0.00001
   bs: 256
   bs_val: 1
-  epochs: 30
+  epochs: 100
   gpu: 0
   l1loss: False
   testing: False
@@ -78,7 +78,4 @@ sub_ico_1:
 
 sub_ico_2:
     num_patches: 320
-    num_vertices: 153 
-
-
-
+    num_vertices: 153 

models/sit.py

@@ -20,7 +20,7 @@
 from einops import repeat
 from einops.layers.torch import Rearrange
 
-from vit_pytorch.vit import Transformer, Attention, FeedForward, PreNorm
+from vit_pytorch.vit import Transformer
 
 class SiT(nn.Module):
     def __init__(self, *,

tools/train.py

@@ -39,14 +39,8 @@
 import numpy as np
 import pandas as pd
 
-from torch.optim.lr_scheduler import StepLR
-from torch.optim.lr_scheduler import ReduceLROnPlateau
-
-
 from models.sit import SiT
 
-from warmup_scheduler import GradualWarmupScheduler
-
 from utils.utils import load_weights_imagenet
 
 from torch.utils.tensorboard import SummaryWriter