chorowski-lab / hCPC

Implementation of multi-level Contrastive Predictive Coding (CPC) methods
MIT License
20 stars 3 forks source link

CPC #4

Open hadas opened 1 year ago

hadas commented 1 year ago

Hi I am trying to run the cpc training but it fails in these lines: captureRes = None if captureOptions != None: for o in captureOptions: assert o in ('pred',) captureRes = {} if 'pred' in captureOptions: assert False # not supported yet, predictions here are in some very weird format it seems captureRes['pred'] = None

in the criterion.py file. I am using the configuration given in the readme. Is it a known issue or I am doing something wrong? Thank you.