karinaodm comments

Repositories
Issues
Comments

Results 3 comments of


                                            karinaodm

Can't use run_segment with apex.amp

I would like to try torch.cuda.amp, but torch.cuda.amp.autocast appears only in PyTorch 1.6 and OptimalGradCheckpointing works only with PyTorch 1.5

Can't use run_segment with apex.amp

Yes, it works with torch.cuda.amp with PyTorch 1.10 after I fixed the line https://github.com/lordfjw/OptimalGradCheckpointing/issues/3#issuecomment-966102707 Thanks!

Problem with PyTorch Version 1.10

I run this code with PyTorch Version 1.10. I replaced https://github.com/lordfjw/OptimalGradCheckpointing/blob/main/graph.py#L162 on `shape = [int(s) for s in shape_str.split(', ') if s.isdigit()]` and get normal output for resnet101 ``` Average...