axel-havard comments

Repositories
Issues
Comments

Results 1 comments of


                                            axel-havard

How about forcing the first and last block on device when groupoffloading is used?

We can flag the first and last layers so that they are not offloaded in the post_forward method. The only issue is that we can only do that through _apply_lazy_group_offloading_hook...