Single view baselines

This commit is contained in:
Michel Breyer
2021-08-25 18:29:10 +02:00
parent fc8d9a8791
commit 5501c2ae42
6 changed files with 89 additions and 207 deletions

View File

@@ -16,7 +16,7 @@ def main():
parser = create_parser()
args = parser.parse_args()
policy = make(args.policy, args.rate, args.filter_grasps)
policy = make(args.policy, args.rate)
controller = GraspController(policy)
logger = Logger(args)
@@ -33,7 +33,6 @@ def create_parser():
parser.add_argument("--runs", type=int, default=10)
parser.add_argument("--logdir", type=Path, default="logs")
parser.add_argument("--rate", type=int, default=5)
parser.add_argument("--filter-grasps", action="store_true")
parser.add_argument("--seed", type=int, default=12)
return parser
@@ -41,11 +40,10 @@ def create_parser():
class Logger:
def __init__(self, args):
stamp = datetime.now().strftime("%y%m%d-%H%M%S")
name = "{}_policy={},rate={},filter-grasps={},seed={}.csv".format(
name = "{}_policy={},rate={},seed={}.csv".format(
stamp,
args.policy,
args.rate,
args.filter_grasps,
args.seed,
)
self.path = args.logdir / name