mirror of
https://github.com/wassname/multifit.git
synced 2026-09-09 11:27:26 +08:00
Fixing Imdb loading
This commit is contained in:
@@ -109,6 +109,7 @@ class CLSHyperParams(LMHyperParams):
|
||||
if 'xnli' in str(self.dataset_dir):
|
||||
NotImplementedError("Support for Xnli is not implemented yet")
|
||||
if 'imdb' in self.dataset_dir.name:
|
||||
lang=''
|
||||
add_trn_to_lm = True
|
||||
if 'mldoc' in str(self.dataset_dir):
|
||||
add_trn_to_lm = False # False as trn_df is contained in unsup already
|
||||
|
||||
Reference in New Issue
Block a user