taha
1 year ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
main.py
View File
| """ | """ | ||||
| model = DeepDRA(cell_sizes, drug_sizes, ae_latent_dim, ae_latent_dim, mlp_input_dim, mlp_output_dim) | |||||
| model = DeepDRA(cell_sizes, drug_sizes, ae_latent_dim, ae_latent_dim) | |||||
| model= model.to(device) | model= model.to(device) | ||||
| # Step 3: Convert your training data to PyTorch tensors | # Step 3: Convert your training data to PyTorch tensors |
Loading…