Shuhao Cao

Results 17 comments of Shuhao Cao

The simplest way is to write your own: ```python def loss1(pred, target): '''first loss here''' return ... def loss2(pred, target): '''second loss here''' return ... def loss(pred, target): return a*loss1(pred,...

https://github.com/TylerYep/torchinfo/blob/ed0a7aa3f61171d2a7f4af8e7958124304c7ef91/torchinfo/layer_info.py#L127-L134 @TylerYep What is the current dev roadmap for this function `calculate_size`? Would you take a PR that adds this functionality supporting dicts containing lists or tuples?

> Thanks for looking into this. Could you provide a small example model + output to illustrate your point? That will help validate this fix works I added a test....

https://github.com/KinglittleQ/torch-batch-svd/blob/c0a96119187f7d55f939d2ff2b92942c6d6ca930/torch_batch_svd/batch_svd.py#L19-L21 Just change this assertion and you are good to go (of course it depends on single/double precision and the VRAM on your GPU).

I wondered this question as well, yet it seems that no one answers questions from months ago #85 The paper did not specify what is the implementation for the interface...