davidaf3
/

ReverseNutrition_TFIngPort

Image Classification

Model card Files Files and versions

davidaf3 commited on Aug 26, 2022

Commit

0832cb7

·

1 Parent(s): 09f2656

Fixed typo

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -12,7 +12,7 @@ import numpy as np
 class PreTrainedPipeline():
     def __init__(self, path=""):
         crop_size = (224, 224)
-        embed_dim = 256,
         num_layers = 3
         seq_length = 20
         hidden_dim = 1024

 class PreTrainedPipeline():
     def __init__(self, path=""):
         crop_size = (224, 224)
+        embed_dim = 256
         num_layers = 3
         seq_length = 20
         hidden_dim = 1024