@@ -77,7 +77,7 @@ to the normal command line arguments, or pass `debug="underflow_overflow"` when
|
|||||||
If you're using your own training loop or another Trainer you can accomplish the same with:
|
If you're using your own training loop or another Trainer you can accomplish the same with:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
from .debug_utils import DebugUnderflowOverflow
|
from transformers.debug_utils import DebugUnderflowOverflow
|
||||||
|
|
||||||
debug_overflow = DebugUnderflowOverflow(model)
|
debug_overflow = DebugUnderflowOverflow(model)
|
||||||
```
|
```
|
||||||
@@ -271,7 +271,7 @@ Additionally, if you're instantiating the debugger in your own code, you can adj
|
|||||||
its default, e.g.:
|
its default, e.g.:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
from .debug_utils import DebugUnderflowOverflow
|
from transformers.debug_utils import DebugUnderflowOverflow
|
||||||
|
|
||||||
debug_overflow = DebugUnderflowOverflow(model, max_frames_to_save=100)
|
debug_overflow = DebugUnderflowOverflow(model, max_frames_to_save=100)
|
||||||
```
|
```
|
||||||
|
|||||||
Reference in New Issue
Block a user