Classes | Functions
my_callbacks Namespace Reference

Classes

class  InceptionV3Callback
 
class  InceptionV4Callback
 
class  IterationsCallback
 
class  ModelCheckpointDetached
 
class  MultiGPUCheckpointCallback
 
class  MyCallback
 

Functions

def detachmodel (m)
 

Function Documentation

def my_callbacks.detachmodel (   m)
Detach model trained on GPUs from its encapsulation
# Arguments
    :param m: obj, keras model
# Returns
    :return: obj, keras model

Definition at line 78 of file my_callbacks.py.

78 def detachmodel(m):
79  """ Detach model trained on GPUs from its encapsulation
80  # Arguments
81  :param m: obj, keras model
82  # Returns
83  :return: obj, keras model
84  """
85 
86  for l in m.layers:
87  if l.name == 'resnext':
88  return l
89 
90  return m
91 
def detachmodel(m)
Definition: my_callbacks.py:78