ryul99 / pytorch-project-template

Deep Learning project template for PyTorch (multi-gpu training is supported)

ryul99/pytorch-project-template Issues

Make member functions of model wrapper to be able to use without feed_data
Closed 2 years ago
Remove loss_f at model.py
Updated 2 years ago
Adding docstrings / type notation for functions
Updated 3 years ago
Cover cuda gpu device notation (e.g. cuda:1)
Updated 3 years ago
Clean up import statements
Closed 3 years ago1
Rename some confusing names
Closed 4 years ago
Use Hydra config
Closed 4 years ago