-
Notifications
You must be signed in to change notification settings - Fork 6
Description
Hello,
Thank you for sharing the codebase! I’ve made a few modifications to the worker code:
1.Added TensorBoard to track the training process.
2.Updated the code so that training can resume if interrupted unexpectedly or stopped for any other reason.
3.Implemented parallel computation for each validation case, improving the overall validation efficiency.
However, I encountered a couple of issues:
My code isn’t utilizing the GPU for training and only uses the CPU. This isn't a critical issue since it doesn't affect the algorithm’s functionality, but I wanted to point it out.I don't know why it's happening.torch.cuda.is_available() showing True
The valid_set folder contains only three files (40length_16agents_0.3density.pth, 40length_32agents_0.3density.pth, and 40length_64agents_0.3density.pth) used for validation. I wanted to know if there is a way to generate other test environments defined in test_env_settings in the config file, which are not present in the folder.
Any guidance on these points would be greatly appreciated!
Thanks again for your help!