added direct links to docs

This commit is contained in:
William Falcon
2020-01-21 13:48:40 -05:00
parent 792706a345
commit b3c51b89d5
2 changed files with 6 additions and 2 deletions
+1 -1
View File
@@ -27,4 +27,4 @@ This is the basic use of the trainer:
from .trainer import Trainer
__all__ = ['Trainer']
__all__ = ['Trainer']
+5 -1
View File
@@ -1,6 +1,7 @@
import os
import sys
import warnings
import logging
import torch
import torch.distributed as dist
@@ -46,7 +47,9 @@ class Trainer(TrainerIOMixin,
TrainerTrainLoopMixin,
TrainerCallbackConfigMixin,
):
r"""Abstract base class used to build new callbacks.
r"""
Main trainer class
"""
def __init__(
@@ -417,6 +420,7 @@ class Trainer(TrainerIOMixin,
"""
#
# .. warning:: Following arguments become deprecated and they will be removed in v0.8.0:
# - `nb_sanity_val_steps`
# Transfer params