agentnet.utils.persistence

from agentnet.utils.persistence import save,load
def build_pretrained_spaceinvaders(observation_reshape):
    """ a smaller version of DQN pre-trained for ~2 hours on GPU """
    assert tuple(observation_reshape.output_shape) == (None, 3, 105, 80)
    
    #main neural network body
    conv0 = Conv2DLayer(observation_reshape,16,filter_size=(8,8),stride=(4,4),name='conv0')
    conv1 = Conv2DLayer(conv0,32,filter_size=(4,4),stride=(2,2),name='conv1')
    dense0 = DenseLayer(conv1,256,name='dense',nonlinearity=lasagne.nonlinearities.tanh)
    
    load(dense0,"./spaceinvaders_dqn_body_small.pcl")
    return dense0