Pj12 commited on
Commit
d9c4aa8
·
verified ·
1 Parent(s): 91a578e

Update extract_feature_print.py

Browse files
Files changed (1) hide show
  1. extract_feature_print.py +1 -1
extract_feature_print.py CHANGED
@@ -131,7 +131,7 @@ class Config:
131
  self.gpu_mem = torch.cuda.get_device_properties(i_device).total_memory // (
132
  1024**3
133
  )
134
- Config()
135
 
136
  #HuggingFacePlaceHolder = None
137
  class HubertModelWithFinalProj(HubertModel):
 
131
  self.gpu_mem = torch.cuda.get_device_properties(i_device).total_memory // (
132
  1024**3
133
  )
134
+ config = Config()
135
 
136
  #HuggingFacePlaceHolder = None
137
  class HubertModelWithFinalProj(HubertModel):