PYTORCH_UTILS# pytorch_utils.set_pytorch_threads() pytorch_utils.set_pytorch_seed(seed) Set the random number seed to the desired value pytorch_utils.get_pytorch_function(name) pytorch_utils.build_pytorch_activation(...) pytorch_utils.build_pytorch_optimizer(model, ...) pytorch_utils.pytorch_initialize(weights, ...) pytorch_utils.pytorch_xent(y_true, y_pred) pytorch_utils.pytorch_mse(y_true, y_pred)
