Update extract_feature_print.py
Browse files- 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):
|