device

This is summary documentation for hippynn.experiment.device module. Click here for full documentation.

Links to Full Documentation:

This file is used to host the function set_devices, as it is needed by both serialization and routines. We may find a better place for this function in the future.

Functions

set_devices(model, loss, evaluator, ...)

Sets the model, loss, and optimizer to the specified device if necessary.