Errors

exception metatrain.utils.errors.ArchitectureError(exception: Exception)

Bases: Exception

Exception raised for errors originating from architectures

This exception should be raised when an error occurs within an architecture’s operation, indicating that the problem is not directly related to the metatrain infrastructure but rather to the specific architecture being used.

Parameters:

exception (Exception) – The original exception that was caught, which led to raising this custom exception.

exception metatrain.utils.errors.OutOfMemoryError(exception: Exception)

Bases: Exception

Exception raised for out of memory errors.

This exception should be raised when a GPU out of memory error occurs, indicating that the model or batch size is too large for the available GPU memory.

Parameters:

exception (Exception) – The original exception that was caught, which led to raising this custom exception.