/
run.py
41 lines (34 loc) · 1.01 KB
/
run.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# -*- coding: utf-8 -*-
"""
This is implementation of basic SpatioTemporal FIlters in Retina+LGN
"""
import sys
from pyNN import nest
from mozaik.controller import run_workflow, setup_logging
import mozaik
from model import T0_Model
from experiments import create_experiments
from mozaik.storage.datastore import Hdf5DataStore,PickledDataStore
from analysis_and_visualization import perform_analysis_and_visualization
from parameters import ParameterSet
try:
from mpi4py import MPI
except ImportError:
MPI = None
if MPI:
mpi_comm = MPI.COMM_WORLD
MPI_ROOT = 0
logger = mozaik.getMozaikLogger()
print sys.argv
if True:
data_store,model = run_workflow( 'T0', T0_Model, create_experiments )
else:
setup_logging()
data_store = PickledDataStore(
load=True,
parameters=ParameterSet({ 'store_stimuli':False, 'root_directory':'T0_data_____' })
,replace=True
)
logger.info('Loaded data store')
if mpi_comm.rank == MPI_ROOT:
perform_analysis_and_visualization(data_store)