ParamConfig.py

# -*- coding: utf-8 -*-
"""
Parameters setting

Created on Feb 2018

@author: fangshuyang (yfs2016@hit.edu.cn)

"""


####################################################
####             MAIN PARAMETERS                ####
####################################################
SimulateData  = True          # If False denotes training the CNN with SEGSaltData
ReUse         = False         # If False always re-train a network 
DataDim       = [2000,301]    # Dimension of original one-shot seismic data
data_dsp_blk  = (5,1)         # Downsampling ratio of input
ModelDim      = [201,301]     # Dimension of one velocity model
label_dsp_blk = (1,1)         # Downsampling ratio of output
dh            = 10            # Space interval 


####################################################
####             NETWORK PARAMETERS             ####
####################################################
if SimulateData:
    Epochs        = 100       # Number of epoch
    TrainSize     = 1600      # Number of training set
    TestSize      = 100       # Number of testing set
    TestBatchSize = 10
else:
    Epochs        = 50
    TrainSize     = 130      
    TestSize      = 10       
    TestBatchSize = 1
    
BatchSize         = 10        # Number of batch size
LearnRate         = 1e-3      # Learning rate
Nclasses          = 1         # Number of output channels
Inchannels        = 29        # Number of input channels, i.e. the number of shots
SaveEpoch         = 20        
DisplayStep       = 2         # Number of steps till outputting stats