DrinkingMilktea

Results 3 issues of DrinkingMilktea

albert classification error(Failed copying input tensor from GPU in order to run Identity: GPU sync failed [Op:Identity])

> #tokenization define from bert.tokenization.albert_tokenization import FullTokenizer def createTokenizer(): return FullTokenizer("../albert_base/assets/30k-clean.vocab", spm_model_file="../albert_base/assets/30k-clean.model", do_lower_case=True) > def get_masks(tokens, max_seq_length): """Mask for padding""" if len(tokens)>max_seq_length: #Cutting down the excess length tokens = tokens[0:max_seq_length]...

docker image to multi architecture

I'm try to use uptrain in M3 macbook laptop. But ``` bash run_uptrain.sh ``` it gave me following error message. ``` server-1 | exec /usr/local/bin/python: exec format error ``` I...

enhancement

docker setting for not x86 architecture

## Pull Request Template ### Type of Change Please delete options that are not relevant. - [x] Bug fix (non-breaking change which fixes an issue) - [ ] New feature...