F.interpolate -> TypeError: _upsample_flops_compute() takes 1 positional argument but 5 were given

[johndpope]

this has code you can run to recreate error.
https://github.com/johndpope/MegaPortrait-hack/tree/feat/26-auditflops

return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,

this code is blowing up profiler. is there a way to ignore?

file "/media/2TB/MegaPortrait-hack/train.py", line 357, in

main(config)

File "/media/2TB/MegaPortrait-hack/train.py", line 350, in main

train_base(cfg, Gbase, Dbase, dataloader)

File "/media/2TB/MegaPortrait-hack/train.py", line 153, in train_base

pred_frame = Gbase(source_frame, driving_frame)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl

return self._call_impl(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1582, in _call_impl

result = forward_call(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/media/2TB/MegaPortrait-hack/model.py", line 1077, in forward

w_s2c = self.warp_generator_s2c(Rs, ts, zs, es)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl

return self._call_impl(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1582, in _call_impl

result = forward_call(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/media/2TB/MegaPortrait-hack/model.py", line 911, in forward

w_em_s2c = self.flowfield(zs_sum,adaptive_gamma,adaptive_beta) ### TODO 3: flowfield do not need them (adaptive_gamma,adaptive_beta)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl

return self._call_impl(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1582, in _call_impl

result = forward_call(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/media/2TB/MegaPortrait-hack/model.py", line 451, in forward

x = self.upsample1(x)

^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl

return self._call_impl(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1582, in _call_impl

result = forward_call(*args, **kwargs)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/torch/nn/modules/upsampling.py", line 157, in forward

return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/home/oem/miniconda3/envs/comfyui/lib/python3.11/site-packages/flops_profiler/profiler.py", line 1008, in newFunc

flops, macs = funcFlopCompute(*args, **kwds)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

TypeError: _upsample_flops_compute() takes 1 positional argument but 5 were given

UPDATE
i comment this out for now.
# upsample
F.upsample = _wrapFunc(F.upsample, _upsample_flops_compute)
# F.interpolate = _wrapFunc(F.interpolate, _interpolate_flops_compute) - has problem #13