fix(networks): AffineHead follows the module's dtype, not just device - #9104
Open
tritsystem wants to merge 1 commit into
Open
fix(networks): AffineHead follows the module's dtype, not just device#9104tritsystem wants to merge 1 commit into
tritsystem wants to merge 1 commit into
