'How to load weights with TPU strategy?

with strategy.scope():
    model = transformer(vocab_size=VOCAB_SIZE,
                        num_layers=NUM_LAYERS,
                        units=UNITS,
                        d_model=D_MODEL,
                        num_heads=NUM_HEADS,
                        is_encoder=True,
                        dropout=DROPOUT)
    
    model.load_weights("path")

I get error:

InvalidArgumentError: Unsuccessful TensorSliceReader constructor: Failed to get matching files on path: UNIMPLEMENTED: File system scheme '[local]' not implemented (file: 'path')

Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution	Source

'How to load weights with TPU strategy?

Sources

Related Questions