def main():
# Define environment variables
# Environment()
# Get parameters from arguments
parser = argparse.ArgumentParser(description='Model training')
parser.add_argument('-c',
'--config_path',
type=str,
default=None,
help='Configuration file')
parser.add_argument('-e',
'--exp_name',
type=str,
default=None,
help='Name of the experiment')
parser.add_argument('-s',
'--shared_path',
type=str,
default='/data/module5',
help='Path to shared data folder')
parser.add_argument('-l',
'--local_path',
type=str,
default='/home/master/sufav/results',
help='Path to local data folder')
arguments = parser.parse_args()
assert arguments.config_path is not None, 'Please provide a configuration' \
'path using -c config/pathname' \
' in the command line'
assert arguments.exp_name is not None, 'Please provide a name for the ' \
'experiment using -e name in the ' \
'command line'
# Define the user paths
shared_path = arguments.shared_path
local_path = arguments.local_path
dataset_path = os.path.join(local_path, 'Datasets')
shared_dataset_path = os.path.join(shared_path, 'Datasets')
experiments_path = os.path.join(local_path, getuser(), 'Experiments')
# Note: this should not be used
shared_experiments_path = os.path.join(shared_path, getuser(),
'Experiments')
usr_path = os.path.join('/home/', getuser())
# Load configuration files
configuration = Configuration(arguments.config_path, arguments.exp_name,
dataset_path, shared_dataset_path,
experiments_path, shared_experiments_path,
usr_path)
cf = configuration.load()
# Train /test/predict with the network, depending on the configuration
process(cf)
# Copy result to shared directory
configuration.copy_to_shared()
