Mengxiangtao

icon indicating a website link https://github.com/mxtsecure/mxtsecure.github.io

icon company 山东大学 icon location 青岛 icon location 山大在读研究生，网络空间安全方向

Results 3 comments of Mengxiangtao

Error on tracking

@crankyz hava you solved it ?

The GPU memory will keep increase

https://github.com/EleutherAI/knowledge-neurons/blob/be1a84adafd6e973ab0ee012527ac1988632a5e8/knowledge_neurons/knowledge_neurons.py#L197C9-L197C55 ```python baseline_outputs = self.model(**encoded_input) ``` I think this step mainly make gpu memory keep increase, just change it following: ```python with torch.no_grad(): baseline_outputs = self.model(**encoded_input) ``` of course, the...

Can it be extended to suit my own fine-tuned huggingface model?

Thanks, do I just need to modify the configuration file (config.py)?