def main (_): FLAGS.channels = 1 FLAGS.classes = 1 FLAGS.db = 'luna16.list' FLAGS.val_db = None FLAGS.epoch_steps = 100 FLAGS.ckpt_epochs = 1 FLAGS.val_epochs = 1000 FLAGS.model = "model" FLAGS.rpn_stride = 1 FLAGS.rpn_params = 4 model = Model() aardvark.train(model) pass
def main(_): model = Model() aardvark.train(model) pass
def main(_): setup_params() model = VoxelNet() aardvark.train(model) pass