text: import sys
  sys.path.append('../core')
  
  from ParametricPolicy import ParametricPolicy
  from ParametricPolicyEnsemble import ParametricPolicyEnsemble
  import numpy as np
  import torch
  import dill
  import ipdb
  
  def main():
  
      data_path    = '../data/'
      network_path = '../models/networks/' 
      states = torch.load(data_path+'states_linear.pt')
      inputs = torch.load(data_path+'inputs_linear.pt')
      
      x_dim = states.shape[1]
      u_dim = inputs.shape[1]
      u_max = 1
      u_min = -1
      
      net_count = 10
      layers    = 5
      hidden_size = 100
  
      policy = ParametricPolicyEnsemble(net_count,x_dim, u_dim, u_max, u_min, layers, hidden_size) 
      policy.train(states,inputs)
      torch.save(policy, network_path+'policy_linear_ensemble.pkl') 
  if __name__ == '__main__':
      main()
  