mirror of
https://github.com/wassname/multifit.git
synced 2026-10-29 13:20:17 +08:00
Fixing Imdb loading
This commit is contained in:
1 parent
119417fb6e
commit
c3276da062
1 file changed
+1
@@ -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