Cleanup GPU Device at process start, at termination and at preemption due to timeout. Log epoch in eval. - Githubissues

rbalestr-lab / stable-SSL

https://rbalestr-lab.github.io/stable-SSL.github.io/dev/

MIT License

4 stars 2 forks source link

Cleanup GPU Device at process start, at termination and at preemption due to timeout. Log epoch in eval. #92

Closed vipulSharma18 closed 2 weeks ago

vipulSharma18 commented 2 weeks ago

GPU and CUDA OOM resolution.
Epoch to be added as a key in eval logging.