Skip to content

Tensors to device

Lupo Pasini, Massimiliano requested to merge tensors_to_device into master
  • tensor.to(device) used in "modules/NN_models.py" and "modules/optimizers.py" to guarantee that GPUs are used if present
  • number of runs introduced as parameter in input file. Each run fixed the random seed to the number of the iteration index
  • nonlinear synthetic dataset added in the "utils/dataloader.py"
  • LeakyReLU added as optional activation function to choose
Edited by Lupo Pasini, Massimiliano

Merge request reports