CPU usage #48
Description
a quick performance monitoring indicated that the bottleneck was concentrated on 1 of my CPU
by profiling the go2_train.py example i get that the 2 slowest functions are torch.as_tensor and _validate_2D_io_variables()
which seem to be called at every step, is it intended ? thanks
the log :
2131668/1608293 2.661 0.000 3.953 0.000 /home/me/Documents/myenv2/lib/python3.12/site-packages/taichi/lang/kernel_impl.py:1214(_getattr)
2400 0.024 0.000 5.266 0.002 /home/me/Documents/rsl_rl/rsl_rl/algorithms/ppo.py:90(act)
4400 10.881 0.002 13.065 0.003 /home/me/Documents/rsl_rl/rsl_rl/modules/actor_critic.py:119(update_distribution)
4400 0.010 0.000 13.644 0.003 /home/me/Documents/rsl_rl/rsl_rl/modules/actor_critic.py:134(act)
83974 22.945 0.000 22.945 0.000 {method 'item' of 'torch._C.TensorBase' objects}
67959 31.006 0.000 31.006 0.000 {built-in method torch.as_tensor}
13558 0.111 0.000 31.402 0.002 /home/me/Documents/Genesis/genesis/engine/solvers/rigid/rigid_solver_decomp.py:3205(_validate_2D_io_variables)