Runtime

Complexity

Total Complexity

5

Size/Duplication

Total Lines	28
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	switch_training()	0	11	3
A	iftrain()	0	6	1
A	__init__()	0	3	1

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import theano
import theano.tensor as T
from theano.ifelse import ifelse
from tensor_conversion import neural_computation

class Runtime(object):
    """
    Manage runtime variables in deepy.
    """

    def __init__(self):
        self._training_flag = theano.shared(0, name="training_flag")
        self._is_training = False


    @neural_computation
    def iftrain(self, then_branch, else_branch):
        """
        Execute `then_branch` when training.
        """
        return ifelse(self._training_flag, then_branch, else_branch, name="iftrain")

    def switch_training(self, flag):
        """
        Switch training mode.
        :param flag: switch on training mode when flag is True.
        """
        if self._is_training == flag: return
        self._is_training = flag
        if flag:
            self._training_flag.set_value(1)
        else:
            self._training_flag.set_value(0)


if "runtime" not in globals():
    runtime = Runtime()