vfdev-5:

Just for info, in Ignite today we also provide a trainer implementation of TBPTT


Here is an updated link

  
      github.com
  
  
    pytorch/ignite/blob/master/ignite/contrib/engines/tbptt.py 167
# coding: utf-8

from enum import Enum

import torch

from ignite._utils import apply_to_tensor
from ignite.engine import Engine, _prepare_batch


class Tbptt_Events(Enum):
    """Aditional tbptt events.

    Additional events for truncated backpropagation throught time dedicated
    trainer.
    """

    TIME_ITERATION_STARTED = "time_iteration_started"
    TIME_ITERATION_COMPLETED = "time_iteration_completed"



  This file has been truncated. show original