AndersonConforto commited on
Commit
d0f5598
·
1 Parent(s): edbf5dc

first commit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ model = HelioSpectFormer(
64
  img_size=224,
65
  patch_size=16,
66
  in_chans=1,
67
- embed_dim=366,
68
  time_embedding={"type": "linear", "time_dim": 1},
69
  depth=8,
70
  n_spectral_blocks=4,
 
64
  img_size=224,
65
  patch_size=16,
66
  in_chans=1,
67
+ embed_dim=368,
68
  time_embedding={"type": "linear", "time_dim": 1},
69
  depth=8,
70
  n_spectral_blocks=4,