Spaces:
Sleeping
Sleeping
ZhaohanM
commited on
Commit
·
ab4771d
1
Parent(s):
feec9dc
Update: SMILES-to-SELFIES conversion, UI polish, and usage guide
Browse files- .ipynb_checkpoints/app-checkpoint.py +3 -1
- app.py +3 -1
.ipynb_checkpoints/app-checkpoint.py
CHANGED
|
@@ -44,7 +44,9 @@ def parse_config():
|
|
| 44 |
p.add_argument("--fusion", default="CAN")
|
| 45 |
p.add_argument("--device", default="cuda" if torch.cuda.is_available() else "cpu")
|
| 46 |
p.add_argument("--save_path_prefix", default="save_model_ckp/")
|
| 47 |
-
p.add_argument("--dataset", default="BindingDB"
|
|
|
|
|
|
|
| 48 |
return p.parse_args()
|
| 49 |
|
| 50 |
args = parse_config()
|
|
|
|
| 44 |
p.add_argument("--fusion", default="CAN")
|
| 45 |
p.add_argument("--device", default="cuda" if torch.cuda.is_available() else "cpu")
|
| 46 |
p.add_argument("--save_path_prefix", default="save_model_ckp/")
|
| 47 |
+
p.add_argument("--dataset", default="BindingDB",
|
| 48 |
+
help="Name of the dataset to use (e.g., 'BindingDB', 'Human', 'Biosnap')")
|
| 49 |
+
|
| 50 |
return p.parse_args()
|
| 51 |
|
| 52 |
args = parse_config()
|
app.py
CHANGED
|
@@ -44,7 +44,9 @@ def parse_config():
|
|
| 44 |
p.add_argument("--fusion", default="CAN")
|
| 45 |
p.add_argument("--device", default="cuda" if torch.cuda.is_available() else "cpu")
|
| 46 |
p.add_argument("--save_path_prefix", default="save_model_ckp/")
|
| 47 |
-
p.add_argument("--dataset", default="BindingDB"
|
|
|
|
|
|
|
| 48 |
return p.parse_args()
|
| 49 |
|
| 50 |
args = parse_config()
|
|
|
|
| 44 |
p.add_argument("--fusion", default="CAN")
|
| 45 |
p.add_argument("--device", default="cuda" if torch.cuda.is_available() else "cpu")
|
| 46 |
p.add_argument("--save_path_prefix", default="save_model_ckp/")
|
| 47 |
+
p.add_argument("--dataset", default="BindingDB",
|
| 48 |
+
help="Name of the dataset to use (e.g., 'BindingDB', 'Human', 'Biosnap')")
|
| 49 |
+
|
| 50 |
return p.parse_args()
|
| 51 |
|
| 52 |
args = parse_config()
|