torch.Tensor or np.ndarray?

automl / Auto-PyTorch

Automatic architecture search and hyperparameter optimization for PyTorch

Apache License 2.0

2.37k stars 289 forks source link

torch.Tensor or np.ndarray? #309

Open nabenabe0928 opened 3 years ago

nabenabe0928 commented 3 years ago

About autoPytorch/pipeline/components/trainer/*.py in refactor_development_regularization_cocktail branch, most data have the typing of np.ndarray, but aren't they torch.Tensor?

ravinkohli commented 3 years ago

they are, we should fix this as well.

nabenabe0928 commented 3 years ago

Actually, there are many mypy error regarding wrong typing expressions, so let's fix after the big merge.

ravinkohli commented 3 years ago

sure.