Question about CPU or GPU error

[chunguangqu]

Question about CPU or GPU error

[ycmin95]

I haven't met this problem before.
It seems like it is about switches between native and cudnn in earlier discussion about pytorch discussion

The recent version of pytorch adopt two different ways for different backends:
For native:
checkAllSameGPU(c, {log_probs_arg, targets_arg});
For cudnn:
checkBackend(c, {*log_probs}, Backend::CUDA); checkBackend(c, {*targets}, Backend::CPU);

You can debug based on your pytorch version and the backend used (torch.backends.cudnn.enabled).

[chunguangqu]

what is your pytorch version?Thanks

[ycmin95]

1.10.0 and 1.10.1 works for me.

[chunguangqu]

My pytorch is also 1.10.0，What is your ctcdecode version?
When I run main.py, the following error occurs:
(your) (base) quchunguang@ubuntu:~/sunday/CSLR$ python main.py
Loading model
Traceback (most recent call last):
File "main.py", line 207, in
processor = Processor(args)
File "main.py", line 33, in init
self.model, self.optimizer = self.loading()
File "main.py", line 96, in loading
loss_weights=self.arg.loss_weights,
File "/home/quchunguang/sunday/CSLR/slr_network.py", line 38, in init
self.decoder = utils.Decode(gloss_dict, num_classes, 'beam')
File "/home/quchunguang/sunday/CSLR/utils/decode.py", line 19, in init
self.ctc_decoder = ctcdecode.CTCBeamDecoder(vocab, beam_width=10, blank_id=blank_id,
AttributeError: module 'ctcdecode' has no attribute 'CTCBeamDecoder'

[ycmin95]

The version has been presented in Readme, it seems like you did not install ctcdecoder successfully.