Merge branch 'demo'

BattModels · Jul 24, 2024 · d5ad54b · d5ad54b
2 parents b99047d + 9125a08
commit d5ad54b
Show file tree

Hide file tree

Showing 7 changed files with 217 additions and 355 deletions.
diff --git a/mist/data_modules/__init__.py b/mist/data_modules/__init__.py
@@ -0,0 +1,2 @@
+from .property_prediction_dataset import PropertyPredictionDataModule
+from .roberta_dataset import RobertaDataSet
diff --git a/mist/models/__init__.py b/mist/models/__init__.py
@@ -0,0 +1,2 @@
+from .lm_finetuning import LMFinetuning
+from .roberta_base import RoBERTa
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		from .property_prediction_dataset import PropertyPredictionDataModule
		from .roberta_dataset import RobertaDataSet
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		from .lm_finetuning import LMFinetuning
		from .roberta_base import RoBERTa