150 epoch:
INFO:waveode_1-rectified_flow:Train Epoch: 150 [6900/6912 (100%)] Loss: 0.004822 time: 0.915s steps: 86975
INFO:waveode_1-rectified_flow:Train Epoch: 150 [0/50 (0%)] Loss: 0.006849
INFO:waveode_1-rectified_flow:Train Epoch: 150 [12/50 (25%)] Loss: 0.009319
INFO:waveode_1-rectified_flow:Train Epoch: 150 [24/50 (50%)] Loss: 0.045419
INFO:waveode_1-rectified_flow:Train Epoch: 150 [36/50 (75%)] Loss: 0.025869
151 epoch:
INFO:waveode_1-rectified_flow:Train Epoch: 151 [6900/6912 (100%)] Loss: 0.004158 time: 0.922s steps: 87551
INFO:waveode_1-rectified_flow:Train Epoch: 151 [0/50 (0%)] Loss: 0.012941
INFO:waveode_1-rectified_flow:Train Epoch: 151 [12/50 (25%)] Loss: 0.002063
INFO:waveode_1-rectified_flow:Train Epoch: 151 [24/50 (50%)] Loss: 0.007402
INFO:waveode_1-rectified_flow:Train Epoch: 151 [36/50 (75%)] Loss: 0.006700