opacus/grad_sample/grad_sample_module.py (1 line): - line 36: # TODO: maybe set max_batch_len on a parameter level, so opacus/validators/module_validator.py (1 line): - line 53: # TODO: use module name here - it's useful part of error message opacus/accountants/utils.py (1 line): - line 38: # TODO: rewrite method to accept GDP opacus/optimizers/optimizer.py (1 line): - line 124: # TODO: handle device transfers: generator and reference tensor website/sphinx/source/conf.py (1 line): - line 41: version = ".".join(splits[:2] + [splits[2][:1]]) # TODO: be smarter here