0.0, 0.6, explore_wt=0.3, enforce_limits=True, sample_fn="uniform" "model.train_aug_stack.transforms.0.cd_rate": HyperParam( 0.01, 0.99, explore_wt=0.3, enforce ...
This needs to be installed according to your CUDA version. Please refer to the official PyTorch installation guide. The following example uses CUDA 12: pip install ...