yychen2000
Results
1
comments of
yychen2000
> compute_amax function is use with np, so it run on cpu, if you want to run fast, you can use torch rewrite it(use torch cuda backend), btw, you can...