iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

try update the api in DataInfo

Browse Source
This commit is contained in:
mhz
2024-06-26 22:10:07 +02:00
parent 0c7c525680
commit 062a27b83f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
graph_dit/main.py
View File

@@ -78,7 +78,7 @@ def main(cfg: DictConfig):
datamodule = dataset.DataModule(cfg) datamodule = dataset.DataModule(cfg)
datamodule.prepare_data() datamodule.prepare_data()
dataset_infos = dataset.DataInfos(datamodule=datamodule, cfg=cfg) dataset_infos = dataset.DataInfos(datamodule=datamodule, cfg=cfg, dataset=datamodule.dataset)
# train_smiles, reference_smiles = datamodule.get_train_smiles() # train_smiles, reference_smiles = datamodule.get_train_smiles()
# get input output dimensions # get input output dimensions