Traceback (most recent call last):
File "main.py", line 113, in <module>
model.train(args.epoch, dataset_train, callbacks=[ckpoint_cb, LossMonitor()], dataset_sink_mode=False)
File "C:\Users\yyyy\Anaconda3\envs\mindspore\lib\site-packages\mindspore\train\model.py", line 535, in train
dataset_sink_mode=dataset_sink_mode)
File "C:\Users\yyyy\Anaconda3\envs\mindspore\lib\site-packages\mindspore\train\model.py", line 357, in _train
self._train_process(epoch, train_dataset, list_callback, cb_params)
File "C:\Users\yyyy\Anaconda3\envs\mindspore\lib\site-packages\mindspore\train\model.py", line 468, in _train_process
outputs = self._train_network(*next_element)
File "C:\Users\yyyy\Anaconda3\envs\mindspore\lib\site-packages\mindspore\nn\cell.py", line 212, in __call__
out = self.compile_and_run(*inputs)
File "C:\Users\yyyy\Anaconda3\envs\mindspore\lib\site-packages\mindspore\nn\cell.py", line 411, in compile_and_run
_executor.compile(self, *inputs, phase=self.phase, auto_parallel_mode=self._auto_parallel_mode)
File "C:\Users\yyyy\Anaconda3\envs\mindspore\lib\site-packages\mindspore\common\api.py", line 417, in compile
result = self._executor.compile(obj, args_list, phase, use_vm)
RuntimeError: mindspore\ccsrc\session\anf_runtime_algorithm.cc:555 GetOutputDeviceDataType] Node [kernel_graph_0:equivdout{[0]: ValueNode<Primitive> FusedBatchNorm, [1]: equivdout, [2]: conv1BatchNorm.gamma, [3]: conv1BatchNorm.beta, [4]: conv1BatchNorm.moving_mean, [5]: conv1BatchNorm.moving_variance}] has a invalid dtype
Works well without the batchnorm layer. Not sure if this is the issue on windows only, will test it on ubuntu16.04.