diff --git a/main.py b/main.py index da1e0c1..5cc73b5 100644 --- a/main.py +++ b/main.py @@ -11,7 +11,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") #for concurrent runs and logging -experiment='RL-baseline-v5' +experiment='RL-baseline-v5-exp4' if __name__ == "__main__": hyperparams = { @@ -23,7 +23,7 @@ 'device': device, 'experiment':experiment, 'params_path': f'./params/policy-params-{experiment}.dl', - 'action_set_num': 0, + 'action_set_num': 3, 'train': True } diff --git a/params/policy-params-RL-baseline-v5-exp4.dl b/params/policy-params-RL-baseline-v5-exp4.dl new file mode 100644 index 0000000..0ea0734 Binary files /dev/null and b/params/policy-params-RL-baseline-v5-exp4.dl differ diff --git a/runs/Apr03_17-13-51_kenya/events.out.tfevents.1617462831.kenya.205693.0 b/runs/Apr03_17-13-51_kenya/events.out.tfevents.1617462831.kenya.205693.0 new file mode 100644 index 0000000..ca0700c Binary files /dev/null and b/runs/Apr03_17-13-51_kenya/events.out.tfevents.1617462831.kenya.205693.0 differ