Example #1
0
   help='Number of neurons per layer')
   parser.add_argument("--epochsSolution", dest="epochsSolution", type=int, default=500,
   help='Number of epochs for pretraining')
   parser.add_argument("--epochsPDE", dest="epochsPDE", type=int, default=3500,
   help='Number of epochs for PDE training')
   parser.add_argument("--pretraining", dest="pretraining", type=int, default=1,
   help='Deactivates(0) or Activates(1) the pretraining step")
   parser.add_argument("--alpha",dest="alpha",type=float, default=12,
   help='Weighting for the initial state')
   parser.add_argument("--lhs",dest="lhs",type=int,default=1,
   help='Deactivates(0) or Activates(1) the lhs sampling')
   parser.add_argument("--pdata",dest="pdata", type=str, default='data/',
   help='Path to initial data')
   parser.add_argument("--modelpath",dest='modelpath',type=str, default='models/',
   help='Path for model saving')
   parser.add_argment("--tensorboard",dest='tensorboard',type=str, default='tensorboard/',
   help='Path for tensoarboard logging')
   parser.add_argment("--nx",dest='nx',type=int, default=200,
   parser.add_argment("--ny",dest='ny',type=int, default=200)
   parser.add_argment("--xmin",dest='xmin',type=float, default=-3.)
   parser.add_argment("--xmax",dest='xmax',type=float, default=+3.)
   parser.add_argment("--ymin",dest='ymin',type=float, default=-3.)
   parser.add_argment("--ymax",dest='ymax',type=float, default=+3.)
   parser.add_argment("--dt",dest='dt',type=float, default=0.001)
   parser.add_argment("--nt",dest='nt',type=int, default=1000)
   parser.add_argment("--integral_x",dest='sampling_x',type=int, default=100)
   parser.add_argment("--integral_y",dest='sampling_y',type=int, default=100)
   parser.add_argument("--lr_solution",dest="lr_solution",type=float,default=3e-5,
   help='Learning Rate for pretraining')
   parser.add_argument("--lr_pde",dest="lr_pde",type=float,default=9e-6,
   help='Learning Rate for pde raining')