diff --git a/main.py b/main.py index da1e0c1..2f4b77c 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-exp2' 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': 1, 'train': True } diff --git a/params/policy-params-RL-baseline-v5-exp2.dl b/params/policy-params-RL-baseline-v5-exp2.dl new file mode 100644 index 0000000..f64e1d6 Binary files /dev/null and b/params/policy-params-RL-baseline-v5-exp2.dl differ diff --git a/runs/Mar31_12-15-11_kenya/events.out.tfevents.1617185711.kenya.87819.0 b/runs/Mar31_12-15-11_kenya/events.out.tfevents.1617185711.kenya.87819.0 new file mode 100644 index 0000000..535b3c9 Binary files /dev/null and b/runs/Mar31_12-15-11_kenya/events.out.tfevents.1617185711.kenya.87819.0 differ