RotoGrad
RotoGrad API
RotateOnly
RotateOnly.num_tasks
RotateOnly.backbone
RotateOnly.heads
RotateOnly.rep
RotateOnly.backward()
RotateOnly.forward()
RotateOnly.rotation
RotateOnly.to()
RotateOnly.train()
RotoGrad.num_tasks
RotoGrad.backbone
RotoGrad.heads
RotoGrad.rep
RotoGrad.forward()
RotoGrad.backward()
RotoGrad.rotation
RotoGradNorm
RotoGradNorm.num_tasks
RotoGradNorm.backbone
RotoGradNorm.heads
RotoGradNorm.rep
RotoGradNorm.forward()
RotoGradNorm.backward()
RotoGradNorm.rotation
RotoGradNorm.weight
cached()
Context-manager that enables the caching system (used for avoid recomputing rotation matrices).