get_precision_context#
- composer.core.get_precision_context(precision, precision_config=None)[source]#
Returns a context manager to automatically cast to a specific precision.
- Parameters
precision_config (Optional[Dict[str, Any]]) โ Config for FP8 scaling strategy. See parameters for DelayedScaling.